Commit
·
b4020b9
1
Parent(s):
f49ac6f
Add logging
Browse filesSigned-off-by: Aivin V. Solatorio <avsolatorio@gmail.com>
- 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 |
-
|
127 |
{"role": "user", "content": message},
|
128 |
{
|
129 |
"role": "assistant",
|
130 |
"content": "Please connect to an MCP server first.",
|
131 |
},
|
132 |
-
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
133 |
|
134 |
-
|
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]]
|