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

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

Loading…
Cancel
Save