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 |
+
|