wehub-resource-sync
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
from langchain.agents import create_agent
|
|
from langchain_core.messages import HumanMessage
|
|
from langchain_openai import ChatOpenAI
|
|
from deepeval.integrations.langchain import tool, CallbackHandler
|
|
from deepeval.prompt import Prompt
|
|
import os
|
|
from tests.test_integrations.utils import (
|
|
assert_trace_json,
|
|
)
|
|
|
|
prompt = Prompt(alias="asd")
|
|
prompt._version = "00.00.01"
|
|
prompt.label = "test-label"
|
|
prompt.hash = "bab04ec"
|
|
|
|
|
|
@tool(metric_collection="test_collection_1")
|
|
def multiply(a: int, b: int) -> int:
|
|
"""Returns the product of two numbers"""
|
|
return a * b
|
|
|
|
|
|
llm = ChatOpenAI(
|
|
model="gpt-4o-mini",
|
|
metadata={"metric_collection": "test_collection_1", "prompt": prompt},
|
|
)
|
|
|
|
agent_executor = create_agent(
|
|
llm,
|
|
[multiply],
|
|
system_prompt="You are a helpful assistant that can perform mathematical operations.",
|
|
)
|
|
|
|
|
|
_current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
json_path = os.path.join(_current_dir, "langchain.json")
|
|
|
|
|
|
# @generate_trace_json(json_path)
|
|
@assert_trace_json(json_path)
|
|
def test_execute_agent():
|
|
agent_executor.invoke(
|
|
{"messages": [HumanMessage(content="What is 8 multiplied by 6?")]},
|
|
config={
|
|
"callbacks": [CallbackHandler(metric_collection="task_completion")]
|
|
},
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_execute_agent()
|