debug

Queue_Producer.py

@@ -0,0 +1,31 @@
+# Producer
+import pika
+import os
+RABBITMQ_URL = os.environ["RABBITMQ_URL"]
+QUEUE_NAME = os.environ["QUEUE_NAME"] 
+
+def get_connection():
+    params = pika.URLParameters(RABBITMQ_URL)
+    return pika.BlockingConnection(params)
+
+connection = get_connection()
+def send_message(data: dict):
+    connection = get_connection()
+    channel = connection.channel()
+
+    channel.queue_declare(queue=QUEUE_NAME, durable=True)
+
+    message = str(data)
+
+    channel.basic_publish(
+        exchange='',
+        routing_key=QUEUE_NAME,
+        body=message,
+        properties=pika.BasicProperties(
+            delivery_mode=2,  # make message persistent
+        )
+    )
+
+    connection.close()
+
+    return {"status": "sent", "data": data}

app.py

@@ -1,3 +1,5 @@
+from urllib import request
+
 from fastapi import FastAPI
 import uvicorn
 import sys
@@ -8,13 +10,19 @@ from Server import get_response
 from pydantic import BaseModel
 from agent.agent_graph.StateTasks import ProblemState
 import subprocess
+from Queue_Producer import send_message
 
-
-# Start consumer in a separate process
+##################################################
+# START CONSUMERS in a separate process
+##################################################
 for i in range(3):  # Start 3 consumers
     subprocess.Popen(['python','-u','Consumer.py'])
 
 
+##################################################
+# START API
+##################################################
+
 # Create app instance
 app = FastAPI()
 print("Starting API Server...")
@@ -29,42 +37,15 @@ class RequestModel(BaseModel):
     last_state : str
 
 
+##################################################
+# ROUTES
+##################################################
+
 @app.get("/")
 def read_root():
     return {"message": "Hello From CodeBuddyAI!"}
 
-######################
-# Producer
-import pika
-RABBITMQ_URL = os.environ["RABBITMQ_URL"]
-QUEUE_NAME = os.environ["QUEUE_NAME"] 
-
-def get_connection():
-    params = pika.URLParameters(RABBITMQ_URL)
-    return pika.BlockingConnection(params)
-
-connection = get_connection()
-def send_message(data: dict):
-    connection = get_connection()
-    channel = connection.channel()
-
-    channel.queue_declare(queue=QUEUE_NAME, durable=True)
-
-    message = str(data)
-
-    channel.basic_publish(
-        exchange='',
-        routing_key=QUEUE_NAME,
-        body=message,
-        properties=pika.BasicProperties(
-            delivery_mode=2,  # make message persistent
-        )
-    )
-
-    connection.close()
-
-    return {"status": "sent", "data": data}
-######################
+"""
 def old_call(request: RequestModel):
     # fill with last state 
     try:
@@ -83,11 +64,19 @@ def old_call(request: RequestModel):
 
 
     return {"Data": answer}
-
+"""
     
 @app.post("/call/")
 def call(request: RequestModel):
-    return send_message(request.model_dump())
+    try:
+        state = json.loads(request.last_state) 
+    except Exception:
+        state: ProblemState = {
+            "question": request.prompt,
+            "memory": request.memory
+        }
+
+    return send_message(state)
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)