80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
|
from langchain.agents import Tool
|
||
|
|
from langchain.tools import StructuredTool
|
||
|
|
from langchain.tools.base import ToolException
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
from typing import List, Dict
|
||
|
|
# import jsonref
|
||
|
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
class BaseToolModel:
|
||
|
|
name = "BaseToolModel"
|
||
|
|
description = "Tool Description"
|
||
|
|
|
||
|
|
class ToolInputArgs(BaseModel):
|
||
|
|
"""
|
||
|
|
Input for MoveFileTool.
|
||
|
|
Tips:
|
||
|
|
default control Required, e.g. key1 is not Required/key2 is Required
|
||
|
|
"""
|
||
|
|
|
||
|
|
key1: str = Field(default=None, description="hello world!")
|
||
|
|
key2: str = Field(..., description="hello world!!")
|
||
|
|
|
||
|
|
class ToolOutputArgs(BaseModel):
|
||
|
|
"""
|
||
|
|
Input for MoveFileTool.
|
||
|
|
Tips:
|
||
|
|
default control Required, e.g. key1 is not Required/key2 is Required
|
||
|
|
"""
|
||
|
|
|
||
|
|
key1: str = Field(default=None, description="hello world!")
|
||
|
|
key2: str = Field(..., description="hello world!!")
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def run(cls, tool_input_args: ToolInputArgs) -> ToolOutputArgs:
|
||
|
|
"""excute your tool!"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class BaseTools:
|
||
|
|
tools: List[BaseToolModel]
|
||
|
|
|
||
|
|
|
||
|
|
def get_tool_schema(tool: BaseToolModel) -> Dict:
|
||
|
|
'''转json schema结构'''
|
||
|
|
data = jsonref.loads(tool.schema_json())
|
||
|
|
_ = json.dumps(data, indent=4)
|
||
|
|
del data["definitions"]
|
||
|
|
return data
|
||
|
|
|
||
|
|
|
||
|
|
def _handle_error(error: ToolException) -> str:
|
||
|
|
return (
|
||
|
|
"The following errors occurred during tool execution:"
|
||
|
|
+ error.args[0]
|
||
|
|
+ "Please try again."
|
||
|
|
)
|
||
|
|
|
||
|
|
import requests
|
||
|
|
from loguru import logger
|
||
|
|
def fff(city, extensions):
|
||
|
|
url = "https://restapi.amap.com/v3/weather/weatherInfo"
|
||
|
|
json_data = {"key": "4ceb2ef6257a627b72e3be6beab5b059", "city": city, "extensions": extensions}
|
||
|
|
logger.debug(f"json_data: {json_data}")
|
||
|
|
res = requests.get(url, params={"key": "4ceb2ef6257a627b72e3be6beab5b059", "city": city, "extensions": extensions})
|
||
|
|
return res.json()
|
||
|
|
|
||
|
|
|
||
|
|
def toLangchainTools(tools: BaseTools) -> List:
|
||
|
|
''''''
|
||
|
|
return [
|
||
|
|
StructuredTool(
|
||
|
|
name=tool.name,
|
||
|
|
func=tool.run,
|
||
|
|
description=tool.description,
|
||
|
|
args_schema=tool.ToolInputArgs,
|
||
|
|
handle_tool_error=_handle_error,
|
||
|
|
) for tool in tools
|
||
|
|
]
|