Weave

app.py

@@ -14,16 +14,19 @@ from gtts import gTTS
 from io import BytesIO
 from together import Together
 import time  # For delay during index readiness check
+import weave
 
 # Pinecone and OpenAI setup
 pinecone_api_key = os.getenv("PINECONE_API_KEY")
 together_api_key = os.getenv("Together_ai_API")
 openai.api_key = os.getenv("OpenAI_API")
 master_prompt = os.getenv("MasterPrompt")
+WANDB_API_KEY = os.getenv("WANDB_API_KEY")
 
 
-# Initialize Pinecone client
+# Initialize Pinecone client and weave
 pc = Pinecone(api_key=pinecone_api_key)
+weave.init('NaturalFarmingChatbot')
 
 # Create or retrieve Pinecone index
 index_name = "farming-assistant"
@@ -51,6 +54,7 @@ denial_response = "Database scraping is not permitted. Please abide by the terms
 # Initialize Together AI client
 client = Together(api_key=together_api_key)  # Updated Together client initialization
 
+@weave.op()
 def generate_response(question):
     # Generate a response using Together AI
     response = client.chat.completions.create(