remove superfluous changes

This commit is contained in:
knielsen404
2025-07-16 14:26:30 +02:00
parent 094fd6e94e
commit e279b351ed
3 changed files with 7 additions and 119 deletions
+1 -4
View File
@@ -1,6 +1,3 @@
# Note:
This branch is only for testing. Use the chatbot in `chatbot_testing_bundler.py` to test out the setup quickly.
# Invariant Gateway
**LLM proxy to observe and debug what your AI agents are doing.**
@@ -398,4 +395,4 @@ To run a subset of the integration tests, execute:
```bash
bash run.sh integration-tests open_ai/test_chat_with_tool_call.py
```
```
-111
View File
@@ -1,111 +0,0 @@
from httpx import Client
from openai import OpenAI
import os
class SimpleChatbot:
def __init__(self):
self.invariant_authorization_token = os.getenv('INVARIANT_API_KEY')
if not self.invariant_authorization_token:
raise ValueError("INVARIANT_API_KEY is not set")
self.client = OpenAI(
http_client=Client(
headers={
"Invariant-Authorization": f"Bearer {self.invariant_authorization_token}"
},
),
base_url="http://localhost:8005/api/v1/gateway/temp_dataset/openai",
)
self.conversation_history = []
def add_message(self, role, content):
"""Add a message to the conversation history"""
self.conversation_history.append({"role": role, "content": content})
def get_response(self, user_input):
"""Get response from the AI model"""
try:
# Add user message to history
self.add_message("user", user_input)
# Get response from API
response = self.client.chat.completions.create(
model="gpt-4o-mini",
messages=self.conversation_history,
)
# Extract the AI's response
ai_response = response.choices[0].message.content
# Add AI response to history
self.add_message("assistant", ai_response)
return ai_response
except Exception as e:
return f"Error: {str(e)}"
def clear_history(self):
"""Clear the conversation history"""
self.conversation_history = []
print("Conversation history cleared!")
def show_history(self):
"""Display the conversation history"""
if not self.conversation_history:
print("No conversation history yet.")
return
print("\n--- Conversation History ---")
for i, message in enumerate(self.conversation_history, 1):
role = message["role"].capitalize()
content = message["content"]
print(f"{i}. {role}: {content}")
print("--- End History ---\n")
def run(self):
"""Main chatbot loop"""
print("🤖 Simple Chatbot Started!")
print("Commands:")
print(" - Type 'quit' or 'exit' to end the conversation")
print(" - Type 'clear' to clear conversation history")
print(" - Type 'history' to show conversation history")
print(" - Type anything else to chat with the AI")
print("-" * 50)
while True:
try:
user_input = input("\nYou: ").strip()
if not user_input:
continue
# Handle special commands
if user_input.lower() in ['quit', 'exit']:
print("👋 Goodbye!")
break
elif user_input.lower() == 'clear':
self.clear_history()
continue
elif user_input.lower() == 'history':
self.show_history()
continue
# Get AI response
print("AI: ", end="", flush=True)
response = self.get_response(user_input)
print(response)
except KeyboardInterrupt:
print("\n👋 Goodbye!")
break
except Exception as e:
print(f"Unexpected error: {e}")
def main():
chatbot = SimpleChatbot()
chatbot.run()
if __name__ == "__main__":
main()
+6 -4
View File
@@ -3,7 +3,9 @@ POSTGRES_PASSWORD=postgres
POSTGRES_DB=invariantmonitor
POSTGRES_HOST=database
# This now points to the bundler instead of the explorer. Change
# `bundler-session-bundler-1` to the name of the bundler container, if it is different.
INVARIANT_API_URL=http://bundler-session-bundler-1:8210
GUARDRAILS_API_URL=http://bundler-session-bundler-1:8210
# This specifies the Invariant Explorer instance where the gateway will push the traces
# Set this to https://preview-explorer.invariantlabs.ai if you want to push to Preview.
# If you want to push to a local instance of explorer, then specify the app-api docker container name like:
# http://<app-api-docker-container-name>:8000 to push to the local explorer instance.
INVARIANT_API_URL=https://explorer.invariantlabs.ai
GUARDRAILS_API_URL=https://explorer.invariantlabs.ai