camparchimedes commited on
Commit
e85b8cc
·
verified ·
1 Parent(s): a5efc48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py CHANGED
@@ -34,6 +34,34 @@ from langchain_core.messages import BaseMessage
34
  from typing import List
35
  from pydantic import BaseModel, Field
36
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
37
  # -------------------------------=== class setup ===-------------------------------
38
  class InMemoryHistory(BaseChatMessageHistory, BaseModel):
39
  messages: List[BaseMessage] = Field(default_factory=list)
 
34
  from typing import List
35
  from pydantic import BaseModel, Field
36
 
37
+
38
+
39
+
40
+ # -------------------------------=== websocket debug ===-----------------------------
41
+ import websocket
42
+
43
+ def on_message(ws, message):
44
+ print(f"Received: {message}")
45
+
46
+ def on_error(ws, error):
47
+ print(f"Error: {error}")
48
+
49
+ def on_close(ws, close_status_code, close_msg):
50
+ print("Closed")
51
+
52
+ def on_open(ws):
53
+ ws.send("Test message")
54
+
55
+ ws = websocket.WebSocketApp(
56
+ "wss://camparchimedes-session-is-disconnected.hf.space/ws/socket.io/",
57
+ on_message=on_message,
58
+ on_error=on_error,
59
+ on_close=on_close,
60
+ )
61
+
62
+ ws.run_forever()
63
+
64
+
65
  # -------------------------------=== class setup ===-------------------------------
66
  class InMemoryHistory(BaseChatMessageHistory, BaseModel):
67
  messages: List[BaseMessage] = Field(default_factory=list)