from langchain import hub
from langchain.agents import AgentExecutor, create_openai_functions_agent
from langchain_community.utilities.infobip import InfobipAPIWrapper
from langchain_core.tools import StructuredTool
from langchain_openai import ChatOpenAI
from pydantic import BaseModel, Field
instructions = "You are a coding teacher. You are teaching a student how to code. The student asks you a question. You answer the question."
base_prompt = hub.pull("langchain-ai/openai-functions-template")
prompt = base_prompt.partial(instructions=instructions)
llm = ChatOpenAI(temperature=0)
class EmailInput(BaseModel):
body: str = Field(description="Email body text")
to: str = Field(description="Email address to send to. Example: email@example.com")
sender: str = Field(
description="Email address to send from, must be 'validemail@example.com'"
)
subject: str = Field(description="Email subject")
channel: str = Field(description="Email channel, must be 'email'")
infobip_api_wrapper: InfobipAPIWrapper = InfobipAPIWrapper()
infobip_tool = StructuredTool.from_function(
name="infobip_email",
description="Send Email via Infobip. If you need to send email, use infobip_email",
func=infobip_api_wrapper.run,
args_schema=EmailInput,
)
tools = [infobip_tool]
agent = create_openai_functions_agent(llm, tools, prompt)
agent_executor = AgentExecutor(
agent=agent,
tools=tools,
verbose=True,
)
agent_executor.invoke(
{
"input": "Hi, can you please send me an example of Python recursion to my email email@example.com"
}
)