mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 15:48:29 +00:00
22 lines
719 B
Python
22 lines
719 B
Python
from typing import Any, Dict, Optional
|
|
|
|
from pydantic import BaseModel as PydanticBaseModel
|
|
from pydantic import Field as PydanticField
|
|
from pydantic.v1 import BaseModel, Field
|
|
|
|
|
|
class ToolCalling(BaseModel):
|
|
tool_name: str = Field(..., description="The name of the tool to be called.")
|
|
arguments: Optional[Dict[str, Any]] = Field(
|
|
..., description="A dictinary of arguments to be passed to the tool."
|
|
)
|
|
|
|
|
|
class InstructorToolCalling(PydanticBaseModel):
|
|
tool_name: str = PydanticField(
|
|
..., description="The name of the tool to be called."
|
|
)
|
|
arguments: Optional[Dict[str, Any]] = PydanticField(
|
|
..., description="A dictinary of arguments to be passed to the tool."
|
|
)
|