Update app.py
Browse files
app.py
CHANGED
|
@@ -74,7 +74,8 @@ class SpyAgent(BasicAgent):
|
|
| 74 |
|
| 75 |
def llm_caller(self, prompt):
|
| 76 |
client = OpenAI(
|
| 77 |
-
api_key=os.getenv('API_KEY')
|
|
|
|
| 78 |
)
|
| 79 |
completion = client.chat.completions.create(
|
| 80 |
model=os.getenv('MODEL_NAME'),
|
|
|
|
| 74 |
|
| 75 |
def llm_caller(self, prompt):
|
| 76 |
client = OpenAI(
|
| 77 |
+
api_key=os.getenv('API_KEY'),
|
| 78 |
+
base_url=os.getenv('BASE_URL')
|
| 79 |
)
|
| 80 |
completion = client.chat.completions.create(
|
| 81 |
model=os.getenv('MODEL_NAME'),
|