zishan111 commited on
Commit
38fb7c5
·
verified ·
1 Parent(s): 47c840a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -3
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
- gpt-4o-mini')))
111
- agent_builder.start()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+