fix: universal chat when default model invalid (#905)

This commit is contained in:
takatost
2023-08-18 16:20:42 +08:00
committed by GitHub
parent edb06f6aed
commit 1d9cc5ca05
8 changed files with 59 additions and 32 deletions

View File

@@ -0,0 +1,25 @@
from datetime import datetime
from typing import Type
from langchain.tools import BaseTool
from pydantic import Field, BaseModel
class DatetimeToolInput(BaseModel):
type: str = Field(..., description="Type for current time, must be: datetime.")
class DatetimeTool(BaseTool):
"""Tool for querying current datetime."""
name: str = "current_datetime"
args_schema: Type[BaseModel] = DatetimeToolInput
description: str = "A tool when you want to get the current date, time, week, month or year, " \
"and the time zone is UTC. Result is \"<date> <time> <timezone> <week>\"."
def _run(self, type: str) -> str:
# get current time
current_time = datetime.utcnow()
return current_time.strftime("%Y-%m-%d %H:%M:%S UTC+0000 %A")
async def _arun(self, tool_input: str) -> str:
raise NotImplementedError()