[fix][remove agent.agent_id and just agent.id

pull/849/head
Kye Gomez 3 weeks ago
parent f21272930f
commit 103952a5c7

@ -46,7 +46,7 @@ chief_medical_officer = Agent(
""",
model_name="gpt-4o-mini",
model_name="gpt-4o-mini",
max_loops=1,
)
@ -70,7 +70,7 @@ virologist = Agent(
* Secondary condition codes
Document all findings using proper medical coding standards and include rationale for code selection.""",
model_name="gpt-4o-mini",
model_name="gpt-4o-mini",
max_loops=1,
)
@ -95,7 +95,7 @@ internist = Agent(
- Include hierarchical condition category (HCC) codes where applicable
Document supporting evidence for each code selected.""",
model_name="gpt-4o-mini",
model_name="gpt-4o-mini",
max_loops=1,
)
@ -126,7 +126,7 @@ medical_coder = Agent(
3. Symptom Codes
4. Complication Codes
5. Coding Notes""",
model_name="gpt-4o-mini",
model_name="gpt-4o-mini",
max_loops=1,
)
@ -154,7 +154,7 @@ synthesizer = Agent(
- Documentation improvements needed
Include confidence levels and evidence quality for all diagnoses and codes.""",
model_name="gpt-4o-mini",
model_name="gpt-4o-mini",
max_loops=1,
)

@ -279,7 +279,6 @@ class Agent:
def __init__(
self,
agent_id: Optional[str] = agent_id(),
id: Optional[str] = agent_id(),
llm: Optional[Any] = None,
template: Optional[str] = None,
@ -403,7 +402,6 @@ class Agent:
**kwargs,
):
# super().__init__(*args, **kwargs)
self.agent_id = agent_id
self.id = id
self.llm = llm
self.template = template
@ -2270,7 +2268,7 @@ class Agent:
time=time.time(),
tokens=total_tokens,
response=AgentChatCompletionResponse(
id=self.agent_id,
id=self.id,
agent_name=self.agent_name,
object="chat.completion",
choices=ChatCompletionResponseChoice(

Loading…
Cancel
Save