llm/SIA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix client side message handling

Browse Source
This commit is contained in:
geens
2024-11-03 22:44:43 +01:00
parent e243c6ac3b
commit b4c70cd5b8
4 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sia/web_agent.py
View File

@@ -126,7 +126,7 @@ class WebAgent(BaseAgent):
self._set_state(WebAgentState.INFERENCE) self._set_state(WebAgentState.INFERENCE)
self.set_response("") self.set_response("")
response_token_iter = self._llm.infer(self._system_prompt, self.context) response_token_iter = self._llm.infer(f"{self._system_prompt}\n{self._action_schema}", self.context)
response = "" response = ""
for token in response_token_iter: for token in response_token_iter:
response += token response += token

3
sia/web_socket_manager.py
View File

@@ -86,11 +86,8 @@ class WebSocketManager:
}) })
async for msg in ws: async for msg in ws:
if msg.type == WSMsgType.TEXT: if msg.type == WSMsgType.TEXT:
#try:
data = json.loads(msg.data) data = json.loads(msg.data)
await self._handle_client_message(request, data) await self._handle_client_message(request, data)
#except Exception as e:
# print(f"Error handling incoming websocket message: {msg}\n{e}")
elif msg.type == WSMsgType.ERROR: elif msg.type == WSMsgType.ERROR:
print(f"WebSocket connection closed with error: {ws.exception()}") print(f"WebSocket connection closed with error: {ws.exception()}")
finally: finally:

2
system_prompt.md
View File

@@ -85,4 +85,4 @@ By maintaining a dynamic relationship between context and action, SIA can tackle
# User interaction # User interaction
Be a helpful assistant to the user. Be a helpful assistant to the user.
Get to know them and make notes about them. Get to know them and make notes about them.
Open the relevant user notes when you interact with them. Open the relevant user notes when you interact with them.

10
web/src/components/App.jsx
View File

@@ -102,15 +102,15 @@ const SIAInterface = () => {
const handleWebSocketMessage = (message) => { const handleWebSocketMessage = (message) => {
switch (message.type) { switch (message.type) {
case MessageType.STATE_CHANGE: case MessageType.STATE_CHANGE:
setAgentState(message.data); setAgentState(message.state);
break; break;
case MessageType.CONTEXT_UPDATE: case MessageType.CONTEXT_UPDATE:
setContext(message.data); setContext(message.context);
break; break;
case MessageType.RESPONSE_UPDATE: case MessageType.RESPONSE_UPDATE:
setOriginalResponse(message.data); setOriginalResponse(message.response);
setModifiedResponse(message.data); setModifiedResponse(message.response);
setValidationError(message.validation_error || null); setValidationError(message.validation_error);
break; break;
case MessageType.OUTPUT_UPDATE: case MessageType.OUTPUT_UPDATE:
setOutput(prev => prev + message.data); setOutput(prev => prev + message.data);