WhoIsSpyAgentExample

Running

alimamaTech commited on Sep 9, 2024

Commit

dffdc79

verified ·

1 Parent(s): 5b6d419

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,7 +74,8 @@ class SpyAgent(BasicAgent):
     def llm_caller(self, prompt):
         client = OpenAI(
-            api_key=os.getenv('API_KEY')
         )
         completion = client.chat.completions.create(
             model=os.getenv('MODEL_NAME'),

     def llm_caller(self, prompt):
         client = OpenAI(
+            api_key=os.getenv('API_KEY'),
+            base_url=os.getenv('BASE_URL')
         )
         completion = client.chat.completions.create(
             model=os.getenv('MODEL_NAME'),