Updated code to log name and email

app/main.py

@@ -23,18 +23,19 @@ app.add_middleware(
 class ChatInput(BaseModel):
     question: str
     session_id: str
+    name: str
+    email: str
 
 @app.post("/chat")
 async def chat(input: ChatInput):
     print("Got question:", input.question)
     print("Session ID:", input.session_id)
-
     config = {
         'configurable': {
             'thread_id': input.session_id
         }
     }
-    response = await get_response(input.question, config=config)
+    response = await get_response(input.question, name=input.name, email=input.email, config=config)
     return {"answer": response['response']}
 
 
@@ -44,6 +45,8 @@ class FAQInput(BaseModel):
     session_id: str
     question: str
     answer: str
+    name: str
+    email: str
 
 
 #Log FAQ in conversation History even though RAG not being called so future RAG calls have access to info
@@ -52,6 +55,9 @@ async def log_faq(faq: FAQInput):
     session_id = faq.session_id
     question = faq.question
     answer = faq.answer
+    name=faq.name
+    email=faq.email
+
 
     from app.rag import session_histories, HumanMessage, AIMessage, log_chat
     if session_id not in session_histories:
@@ -60,5 +66,5 @@ async def log_faq(faq: FAQInput):
     session_histories[session_id].append(HumanMessage(content=question))
     session_histories[session_id].append(AIMessage(content=answer))
 
-    log_chat(session_id=session_id, query=question, answer=answer, metadata={"source": "FAQ"})
+    log_chat(session_id=session_id, name=name, email=email, query=question, answer=answer, metadata={"source": "FAQ"})
     return {"status": "success"}

app/rag.py

@@ -256,7 +256,7 @@ graph.add_edge("generate_response", END)
 
 app = graph.compile()
 
-async def get_response(query: str, config) -> dict:
+async def get_response(query: str, name, email, config) -> dict:
     start_time = time.time()
     session_id = config['configurable']['thread_id']
     history = session_histories.get(session_id, [])
@@ -276,6 +276,8 @@ async def get_response(query: str, config) -> dict:
 
     log_chat(
         session_id=session_id,
+        name=name,
+        email=email,
         query=query,
         answer=result.get("response", ""),
         latency_ms= latency_ms,