LangChain-Agent自定义Tools类 ——基础篇（一）-Toy模板网

这篇具有很好参考价值的文章主要介绍了LangChain-Agent自定义Tools类 ——基础篇（一）。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

from langchain.agents import Tool
from langchain.tools import BaseTool
from math import pi
from typing import Union
from langchain.agents import initialize_agent
from langchain.agents import AgentType
import os
from langchain.chat_models import ChatOpenAI

os.environ["OPENAI_API_KEY"] = "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
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
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"
llm = ChatOpenAI(
    openai_api_key=os.environ["OPENAI_API_KEY"] ,
    temperature=0,
    model_name='gpt-3.5-turbo'
)
class CustomTool(BaseTool):
    name = "Temperature Detector"
    description = "This is a custom tool for my temperature detection use case"
 
    def _run(self, input: str) -> str:
        # Your logic here
        return "temperature is not bad,huh,20 celceius"
 
    def _arun(self, query: str):
        raise NotImplementedError("This tool does not support async")
class Robotic_Control(BaseTool):
    name = "Robotic Arm Control"
    description = "This is a custom tool for my Robotic Arm Control"
 
    def _run(self, input: str) -> str:
        # Your logic here
        return "Working "
 
    def _arun(self, query: str):
        raise NotImplementedError("This tool does not support async")
tools = [CustomTool(),Robotic_Control()]
 
 
 
# agent = initialize_agent(tools, agent=AgentType.DEFAULT)
agent = initialize_agent(tools, 
                         llm, 
                         agent="zero-shot-react-description", 
                         verbose=True)
agent("How's the temperature and Robotic Arm Control")

LangChain-Agent自定义Tools类 ——基础篇（一）