Integrate OpikTracer into ChatGoogleGenerativeAI and ChatGroq models; remove unused test endpoint and update requirements

app.py

@@ -19,6 +19,7 @@ import groq
 from datetime import datetime
 from fastapi import HTTPException
 from langchain_google_genai import ChatGoogleGenerativeAI
+from opik.integrations.langchain import OpikTracer
 
 # Load environment variables - for local development
 from dotenv import load_dotenv
@@ -142,20 +143,13 @@ def write_message(data: dict, header: str = Header()):
     message = data['message']
     
     try:
-        model = ChatGoogleGenerativeAI(model=GOOGLE_MODEL, max_retries=3)
+        model = ChatGoogleGenerativeAI(model=GOOGLE_MODEL, max_retries=3, callbacks = [OpikTracer()])
     except Exception as e: #fallback model
-        model = ChatGroq(temperature=1, groq_api_key=GROQ_API_KEY, model_name=GROQ_MODEL)
+        model = ChatGroq(model=GROQ_MODEL, temperature=1, callbacks = [OpikTracer()])
     # model = ChatOllama(model="gemma3:1b", temperature=1)
     transaction_bot = Agent(model, system=prompt)
     transaction_bot.graph.invoke({"messages": [message]})
     return {"message": "Transaction completed successfully"}
 
-@app.post("/test")
-def test(header = Header()):
-    if header != HF_TOKEN:
-        raise HTTPException(status_code=400, detail="Invalid header")
-
-    return {"message": "Test successful"}
-
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info")

requirements.txt

@@ -1,9 +1,10 @@
 fastapi
 uvicorn[standard]
-langchain-ollama
 langgraph
 pygsheets
 pandas
 langchain-groq
 dotenv
-langchain-google-genai
+langchain-google-genai
+opik
+langchain