Update app.py

app.py

@@ -1,5 +1,11 @@
 import requests
-from langchain import PromptTemplate, LLMChain
+from langchain import PromptTemplate
+from langsmith import LangSmithClient
+from langsmith.tracker import Tracker
+
+# Initialize LangSmith Client
+client = LangSmithClient(api_key='YOUR_LANGSMITH_API_KEY')  # Replace with your actual LangSmith API key
+tracker = Tracker(client=client)
 
 # Define a simple prompt template
 prompt_template = PromptTemplate(
@@ -11,25 +17,44 @@ prompt_template = PromptTemplate(
 def generate_response(name):
     # Construct the prompt
     prompt = prompt_template.format(name=name)
+    print(f'Constructed Prompt: {prompt}')  # Debugging: Print constructed prompt
     
     # Define API endpoint and headers
-    api_url = 'https://aistudio.google.com/app/apikey?_gl=1*xsxz70*_ga*NDkwNjY1MTA3LjE3MjI5NDUwMjA.*_ga_P1DBVKWT6V*MTcyMjk0NTAxOS4xLjAuMTcyMjk0NTAxOS42MC4wLjY4NzQ4NTc1Mw..'  # Replace with actual Gemini API endpoint
+    api_url = 'https://api.your-google-gemini-endpoint.com/v1/generate'  # Replace with actual Gemini API endpoint
     headers = {
-        'Authorization': f'Bearer YOUR_GEMINI_API_KEY',
+        'Authorization': f'Bearer YOUR_GEMINI_API_KEY',  # Replace with actual Gemini API key
         'Content-Type': 'application/json'
     }
     payload = {
         'prompt': prompt
     }
+    print(f'Sending Request to URL: {api_url}')  # Debugging: Print API URL
+    print(f'Headers: {headers}')  # Debugging: Print headers
+    print(f'Payload: {payload}')  # Debugging: Print payload
 
-    # Send the request to the API
-    response = requests.post(api_url, headers=headers, json=payload)
+    # Track the request using LangSmith
+    tracker.track_event('LLM Request', {'name': name, 'prompt': prompt})
     
-    if response.status_code == 200:
+    # Send the request to the API
+    try:
+        response = requests.post(api_url, headers=headers, json=payload)
+        response.raise_for_status()  # Raise an HTTPError for bad responses (4xx and 5xx)
+        
+        # Debugging: Print raw response
+        print(f'Response Status Code: {response.status_code}')
+        print(f'Response Text: {response.text}')
+        
         result = response.json()
         generated_text = result.get('text', 'No response text available.')
-    else:
-        generated_text = 'Error: Unable to get response from the API.'
+        
+        # Track the response using LangSmith
+        tracker.track_event('LLM Response', {'response': generated_text})
+    except requests.exceptions.RequestException as e:
+        generated_text = f'Error: {e}'
+        print(generated_text)  # Print the error
+        
+        # Track the error using LangSmith
+        tracker.track_event('LLM Error', {'error': str(e)})
 
     return generated_text
 
@@ -39,4 +64,3 @@ if __name__ == "__main__":
     response = generate_response(name)
     print(f"Generated Response: {response}")
 
-