fix(rag): correct dict iteration

Author: sanjayankur31

rag_pkg/gen_rag/api/chat.py

@@ -8,6 +8,8 @@
 Author: Ankur Sinha <sanjay DOT ankur AT gmail DOT com>
 """
 
+import traceback
+
 from fastapi import APIRouter, HTTPException, Request
 
 chat_router = APIRouter()
@@ -19,7 +21,7 @@ async def query(request: Request, query: str):
     try:
         result = await rag.run_graph_invoke(query)
     except Exception as e:
-        print(e)
-        result = HTTPException(status_code=500, detail=str(e))
+        detail = f"{e}\n{traceback.format_exc()}"
+        result = HTTPException(status_code=500, detail=detail)
 
     return {"result": result}

rag_pkg/gen_rag/rag.py

@@ -168,9 +168,9 @@ def _classify_question_domain(self, state: RAGState) -> dict:
 
         domain_str = ""
 
-        for d, info in domain_info:
-            desc = info.get("description", None)
-            if not desc:
+        for d, info in domain_info.items():
+            desc = info.description
+            if not desc or len(desc) == 0:
                 desc = f"if the question is about {d}"
             else:
                 desc = f"if the question is about {desc}"