tool.py 381 B

1234567891011
  1. from litellm import ModelResponse
  2. from pydantic import BaseModel
  3. class ToolCallMetadata(BaseModel):
  4. # See https://docs.litellm.ai/docs/completion/function_call#step-3---second-litellmcompletion-call
  5. function_name: str # Name of the function that was called
  6. tool_call_id: str # ID of the tool call
  7. model_response: ModelResponse
  8. total_calls_in_response: int