Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -106,6 +106,21 @@ class SpyAgent(BasicAgent):
|
|
| 106 |
|
| 107 |
if __name__ == '__main__':
|
| 108 |
name = 'spy'
|
| 109 |
-
agent_builder = AgentBuilder(name, agent=SpyAgent(name, model_name=os.getenv('
|
| 110 |
-
|
| 111 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 106 |
|
| 107 |
if __name__ == '__main__':
|
| 108 |
name = 'spy'
|
| 109 |
+
agent_builder = AgentBuilder(name, agent=SpyAgent(name, model_name=os.getenv('MODEL_NAME')))
|
| 110 |
+
agent_builder.start()
|
| 111 |
+
|
| 112 |
+
|
| 113 |
+
base_url=os.getenv('BASE_URL')
|
| 114 |
+
)
|
| 115 |
+
completion = client.chat.completions.create(
|
| 116 |
+
model=self.model_name,
|
| 117 |
+
messages=[
|
| 118 |
+
{'role': 'system', 'content': 'You are a helpful assistant.'},
|
| 119 |
+
{'role': 'user', 'content': prompt}
|
| 120 |
+
|
| 121 |
+
if __name__ == '__main__':
|
| 122 |
+
name = 'spy'
|
| 123 |
+
agent_builder = AgentBuilder(name, agent=SpyAgent(name, model_name=os.getenv('MODEL_NAME')'))
|
| 124 |
+
agent_builder.start()
|
| 125 |
+
|
| 126 |
+
|