gistfile1.txt

class CohereFunctionModel(BaseModel):
    name: str
    description: str
    parameter_definitions: dict

    @classmethod
    def from_pydantic_model(cls, model: BaseModel):
        return cls(
            name=model.__name__,
            description=model.__doc__,
            parameter_definitions={
                field_name: {
                    "description": field.description,
                    "type": field.annotation.__name__,
                    "required": field.is_required()
                }
                for field_name, field in model.model_fields.items()
            }
        )
        
def function_calling_schema(model):
    """Pydantic to JSON schema for function calling"""
    
    return CohereFunctionModel.from_pydantic_model(model).model_dump()