Upload 2 files

app.py

@@ -1,6 +1,7 @@
 import os
 import streamlit as st
 import instructor
+import httpx
 from atomic_agents.lib.components.agent_memory import AgentMemory
 from atomic_agents.lib.components.system_prompt_generator import SystemPromptGenerator
 from atomic_agents.agents.base_agent import BaseAgent, BaseAgentConfig, BaseAgentInputSchema, BaseAgentOutputSchema
@@ -22,15 +23,19 @@ def setup_client(provider):
         if not api_key:
             st.error("OPENAI_API_KEY not set in environment variables.")
             return None, None, None
-        # Explicitly pass only the required arguments to AsyncOpenAI
-        openai_client = AsyncOpenAI(api_key=api_key)
+        # Create a custom HTTP client without proxies
+        http_client = httpx.AsyncClient()
+        # Initialize AsyncOpenAI with the custom HTTP client
+        openai_client = AsyncOpenAI(api_key=api_key, http_client=http_client)
         client = instructor.from_openai(openai_client)
         model = "gpt-4o-mini"
         display_model = "OpenAI (gpt-4o-mini)"
     elif provider == "ollama":
         from openai import AsyncOpenAI as OllamaClient
-        # Explicitly pass only the required arguments to OllamaClient
-        ollama_client = OllamaClient(base_url="http://localhost:11434/v1", api_key="ollama")
+        # Create a custom HTTP client without proxies
+        http_client = httpx.AsyncClient()
+        # Initialize OllamaClient with the custom HTTP client
+        ollama_client = OllamaClient(base_url="http://localhost:11434/v1", api_key="ollama", http_client=http_client)
         client = instructor.from_openai(ollama_client, mode=instructor.Mode.JSON)
         model = "llama3"
         display_model = "Ollama (llama3)"

requirements.txt

@@ -2,4 +2,5 @@ streamlit==1.38.0
 instructor==1.3.4
 atomic-agents==0.2.1
 python-dotenv==1.0.1
-openai==1.35.12
+openai==1.35.12
+httpx==0.27.0