diff --git a/sanclone/agent/__init__.py b/sanclone/agent/__init__.py new file mode 100644 index 0000000..bf3eda0 --- /dev/null +++ b/sanclone/agent/__init__.py @@ -0,0 +1,3 @@ +from agent import SanCloneAgent + +__all__ = ["SanCloneAgent"] diff --git a/sanclone/agent/agent.py b/sanclone/agent/agent.py new file mode 100644 index 0000000..a86cad6 --- /dev/null +++ b/sanclone/agent/agent.py @@ -0,0 +1,41 @@ +from langchain.agents import AgentExecutor, ZeroShotAgent +from langchain.agents.openai_functions_agent.base import OpenAIFunctionsAgent +from langchain.chat_models import ChatOpenAI + +from ..tools import make_tools +from .prompt import prompt_template + + +class AgentType: + valid_models = { + "ReactAgent": ZeroShotAgent, + "OpenAIFunctionsAgent": OpenAIFunctionsAgent, + } + + @classmethod + def get_agent(cls, model_name: str = "ReactAgent"): + return cls.valid_models[model_name] + + +class SanCloneAgent: + def __init__( + self, + tools=None, + llm=None, + openai_api_key=None, + temp=0.1, + agent_type: str = "OpenAIFunctionsAgent", + verbose=True, + ): + llm = ChatOpenAI(temperature=0.0, model="gpt-4", client=None) + + tools = make_tools(llm) + self.agent_instance = AgentExecutor.from_agent_and_tools( + tools=tools, + agent=AgentType.get_agent(agent_type).from_llm_and_tools(llm, tools), + return_intermediate_steps=True, + handle_parsing_errors=True, + ) + + def run(self, prompt: str): + return self.agent_instance.run(prompt_template.format(input=prompt)) diff --git a/sanclone/agent/prompt.py b/sanclone/agent/prompt.py new file mode 100644 index 0000000..9be2b86 --- /dev/null +++ b/sanclone/agent/prompt.py @@ -0,0 +1,8 @@ +# flake8: noqa +prompt_template = """ +You are an expert in molecular cloning. +You have a set of tools at your disposal. +Your task is to respond to the question or +solve the problem to the best of your ability using the provided tools. +Here is the question: {input} +""" diff --git a/sanclone/tools/__init__.py b/sanclone/tools/__init__.py index 1095f07..816f89e 100644 --- a/sanclone/tools/__init__.py +++ b/sanclone/tools/__init__.py @@ -1,3 +1,4 @@ from .echo import EchoTool +from .maketools import make_tools -__all__ = ["EchoTool"] +__all__ = ["EchoTool", "make_tools"] diff --git a/sanclone/tools/maketools.py b/sanclone/tools/maketools.py new file mode 100644 index 0000000..55ea737 --- /dev/null +++ b/sanclone/tools/maketools.py @@ -0,0 +1,11 @@ +from langchain import agents +from langchain.llms.base import BaseLanguageModel + + +def make_tools(llm: BaseLanguageModel): + # add human input tool + tools = agents.load_tools(["human"], llm) + + # append tools here + tools += [] + return tools