Initial commit for deployment

test_chat_with_params.py

@@ -0,0 +1,69 @@
+"""
+Test script for sending a message with advanced parameters.
+"""
+
+import requests
+import json
+import uuid
+
+# Create a new chat
+def create_chat():
+    response = requests.post(
+        "http://localhost:5251/chats",
+        json={
+            "user_id": "test_user",
+            "title": "Test Chat",
+            "model_id": "gpt-3.5-turbo",
+            "is_team_chat": False
+        }
+    )
+    
+    if response.status_code == 200:
+        return response.json()["id"]
+    else:
+        print(f"Error creating chat: {response.status_code}")
+        print(response.text)
+        return None
+
+# Send a message with advanced parameters
+def send_message(chat_id):
+    response = requests.post(
+        f"http://localhost:5251/chats/{chat_id}/messages",
+        json={
+            "message": "Tell me about artificial intelligence",
+            "user_id": "test_user",
+            "use_rag": False,
+            "temperature": 0.7,
+            "max_tokens": 500,
+            "top_p": 0.9,
+            "frequency_penalty": 0.5,
+            "presence_penalty": 0.5,
+            "system_prompt": "You are a helpful AI assistant that provides concise responses."
+        }
+    )
+    
+    if response.status_code == 200:
+        return response.json()
+    else:
+        print(f"Error sending message: {response.status_code}")
+        print(response.text)
+        return None
+
+# Main function
+def main():
+    print("Creating a new chat...")
+    chat_id = create_chat()
+    
+    if chat_id:
+        print(f"Chat created with ID: {chat_id}")
+        
+        print("\nSending a message with advanced parameters...")
+        response = send_message(chat_id)
+        
+        if response:
+            print("\nResponse received:")
+            print(f"Message ID: {response['id']}")
+            print(f"Content: {response['content']}")
+    
+if __name__ == "__main__":
+    main()