JuyeopDang commited on
Commit
e6ed150
·
verified ·
1 Parent(s): ebccda5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -8
app.py CHANGED
@@ -21,19 +21,14 @@ class BasicAgent:
21
  def __call__(self, question: str) -> str:
22
  print(f"Agent received question (first 50 chars): {question[:50]}...")
23
  fixed_answer = "This is a default answer."
24
- message = [
25
- {
26
- "role": "system",
27
- "content": """You are an agent that finds answers to user queries using web searches and other available resources. Your response must be a concise, direct, and to-the-point answer, ideally a single phrase or sentence. Avoid any elaboration or unnecessary details.""",
28
- },
29
  {
30
  "role": "user",
31
  "content": question
32
- }
33
- ]
34
  completion = self.client.chat.completions.create(
35
  messages=message,
36
- model="compound-beta",
37
  )
38
  answer = completion.choices[0].message.content
39
  print(f"First answer: {answer}")
 
21
  def __call__(self, question: str) -> str:
22
  print(f"Agent received question (first 50 chars): {question[:50]}...")
23
  fixed_answer = "This is a default answer."
24
+ message = [
 
 
 
 
25
  {
26
  "role": "user",
27
  "content": question
28
+ }]
 
29
  completion = self.client.chat.completions.create(
30
  messages=message,
31
+ model="compound-beta-mini",
32
  )
33
  answer = completion.choices[0].message.content
34
  print(f"First answer: {answer}")