Updated Chatbot

Author: dristy-daxa

safe_infer_chatbot_app/safe_infer_chatbot.py

@@ -5,6 +5,8 @@
 from typing import Dict, Any
 import time
 
+ 
+
 # Page configuration
 st.set_page_config(
     page_title="SafeInfer LLM Chatbot",
@@ -98,8 +100,7 @@ def call_safe_infer_api(message: str, model: str, api_key: str = "") -> Dict[str
 
     payload = {
         "model": model,
-        "input": message,
-        "app": "chatbot",
+        "input": message
     }
 
     try:

safe_infer_chatbot_app/utils.py

@@ -2,38 +2,16 @@
 
 import requests
 
-def get_proxima_cookie():
-    try:
-        proxima_url = f"{os.environ.get('PROXIMA_HOST')}/api/auth/login"
-        headers = {
-            'accept': 'application/json',
-            'Content-Type': 'application/x-www-form-urlencoded'
-        }
-        data = {
-            "username": os.environ.get("PROXIMA_USER_USERNAME"),
-            "password": os.environ.get("PROXIMA_USER_PASSWORD")
-        }
-        response = requests.post(proxima_url, headers=headers, data=data)
-
-        response.raise_for_status()
-
-        token = response.headers['set-cookie'].split(';')[0]
-        return token
-    except Exception as e:
-        print(f"Error getting proxima cookie: {e}")
-        raise e
-
-
 
 def get_available_models():
     try:
-        proxima_cookie = get_proxima_cookie()
         headers = {
             'accept': 'application/json',
+            'Authorization': f"Bearer {os.environ.get('PEBBLO_API_KEY')}",
             'Content-Type': 'application/x-www-form-urlencoded',
-            'Cookie': f"fastapiusersauth={proxima_cookie}"
         }
-        response = requests.get(f"{os.environ.get('PROXIMA_HOST')}/api/llm/provider", headers=headers)
+        url = f"{os.environ.get('PROXIMA_HOST')}/safe_infer/llm/provider/list"
+        response = requests.get(url, headers=headers)
         response.raise_for_status()
         models = response.json()
         if len(models) == 0: