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_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 = ""
for token in response_token_iter:
response += token

3
sia/web_socket_manager.py
View File

@@ -86,11 +86,8 @@ class WebSocketManager:
})
async for msg in ws:
if msg.type == WSMsgType.TEXT:
#try:
data = json.loads(msg.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:
print(f"WebSocket connection closed with error: {ws.exception()}")
finally:

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

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