avsolatorio commited on
Commit
b4020b9
·
1 Parent(s): f49ac6f

Add logging

Browse files

Signed-off-by: Aivin V. Solatorio <avsolatorio@gmail.com>

Files changed (1) hide show
  1. mcp_client.py +11 -6
mcp_client.py CHANGED
@@ -123,18 +123,23 @@ class MCPClientWrapper:
123
  self, message: str, history: List[Union[Dict[str, Any], ChatMessage]]
124
  ) -> tuple:
125
  if not self.session:
126
- return history + [
127
  {"role": "user", "content": message},
128
  {
129
  "role": "assistant",
130
  "content": "Please connect to an MCP server first.",
131
  },
132
- ], gr.Textbox(value="")
 
 
 
 
 
 
 
 
133
 
134
- new_messages = loop.run_until_complete(self._process_query(message, history))
135
- return history + [
136
- {"role": "user", "content": message}
137
- ] + new_messages, gr.Textbox(value="")
138
 
139
  async def _process_query(
140
  self, message: str, history: List[Union[Dict[str, Any], ChatMessage]]
 
123
  self, message: str, history: List[Union[Dict[str, Any], ChatMessage]]
124
  ) -> tuple:
125
  if not self.session:
126
+ messages = history + [
127
  {"role": "user", "content": message},
128
  {
129
  "role": "assistant",
130
  "content": "Please connect to an MCP server first.",
131
  },
132
+ ]
133
+ else:
134
+ new_messages = loop.run_until_complete(
135
+ self._process_query(message, history)
136
+ )
137
+ messages = history + [{"role": "user", "content": message}] + new_messages
138
+
139
+ with open("messages.log.jsonl", "a+") as fl:
140
+ fl.write(json.dumps(dict(time=f"{datetime.now()}", messages=messages)))
141
 
142
+ return messages, gr.Textbox(value="")
 
 
 
143
 
144
  async def _process_query(
145
  self, message: str, history: List[Union[Dict[str, Any], ChatMessage]]