NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-29 17:35:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix

Browse source
This commit is contained in:
FIRST_NAME LAST_NAME 2025-05-19 00:46:10 +00:00 • committed by tsadpbb
parent 2acd8aef3e
commit f401a746f1
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
environments/hack0/doctor_agent/doctor.py
View file

@ -203,6 +203,8 @@ class DoctorEnv(BaseEnv):
patient_msg = completion.choices[0].message.content
# print("patient message", patient_msg)
print("patient message", patient_msg)
doctor_messages.append({"role": USER_TAG, "content": patient_msg})
patient_messages.append({"role": ASSISTANT_TAG, "content": patient_msg})
# print("after xai message")
@ -222,12 +224,14 @@ class DoctorEnv(BaseEnv):
# print("before doctor response")
# print("messages", doctor_messages)
doctor_completions = await server.chat_completion(
messages=[{"role" : USER_TAG, "content": "test"}],
messages=doctor_messages,
n=1,
max_tokens=max_tokens,
)
doctor_msg = doctor_completions.choices[0].message.content
print("doctor message", doctor_msg)
# print("doctor message", doctor_msg)
doctor_messages.append({"role": ASSISTANT_TAG, "content": doctor_msg})