Update app.py

app.py

@@ -1,11 +1,12 @@
+# Import necessary libraries
 import requests
-from langchain import PromptTemplate
-from langsmith import LangSmithClient
-from langsmith.tracker import Tracker
+from langchain import PromptTemplate, LLMChain
+# Adjust based on actual class available in the langsmith package
+from langsmith import Client as LangSmithClient
 
 # Initialize LangSmith Client
-client = LangSmithClient(api_key='YOUR_LANGSMITH_API_KEY')  # Replace with your actual LangSmith API key
-tracker = Tracker(client=client)
+client = LangSmithClient(api_key='YOUR_LANGSMITH_API_KEY')
+tracker = client.get_tracker()  # Adjust based on actual method for tracking
 
 # Define a simple prompt template
 prompt_template = PromptTemplate(
@@ -17,44 +18,29 @@ 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://api.your-google-gemini-endpoint.com/v1/generate'  # Replace with actual Gemini API endpoint
     headers = {
-        'Authorization': f'Bearer YOUR_GEMINI_API_KEY',  # Replace with actual Gemini API key
+        'Authorization': f'Bearer YOUR_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
 
-    # Track the request using LangSmith
-    tracker.track_event('LLM Request', {'name': name, 'prompt': prompt})
-    
     # 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.')
-        
-        # 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)})
+
+    # Track the request and response
+    tracker.track_event('LLM Request', {'name': name, 'prompt': prompt})
+    tracker.track_event('LLM Response', {'response': generated_text})
 
     return generated_text
 
@@ -64,3 +50,4 @@ if __name__ == "__main__":
     response = generate_response(name)
     print(f"Generated Response: {response}")
 
+