Fixes for library client (#587)

Library client used _server_ side types which was no bueno. The fix here
is not the completely correct fix but it is good for enough and for the
demo notebook.
This commit is contained in:
Ashwin Bharambe 2024-12-09 17:14:37 -08:00 committed by GitHub
parent 7615da78b8
commit a4d8a6009a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 89 additions and 84 deletions

View file

@ -340,9 +340,8 @@ class AgentTurnResponseStepProgressPayload(BaseModel):
step_type: StepType
step_id: str
model_response_text_delta: Optional[str] = None
text_delta: Optional[str] = None
tool_call_delta: Optional[ToolCallDelta] = None
tool_response_text_delta: Optional[str] = None
@json_schema_type

View file

@ -121,7 +121,7 @@ class EventLogger:
else:
yield event, LogEvent(
role=None,
content=event.payload.model_response_text_delta,
content=event.payload.text_delta,
end="",
color="yellow",
)