From 6c4f04a2a0010a780e6be1517b696ee67141e434 Mon Sep 17 00:00:00 2001 From: Lina Tang Date: Fri, 19 Jan 2024 17:03:16 +0800 Subject: [PATCH] Fix test_batch_with_langchain --- .../flow_with_langchain_traces/test_langchain_traces.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/promptflow/tests/test_configs/flows/flow_with_langchain_traces/test_langchain_traces.py b/src/promptflow/tests/test_configs/flows/flow_with_langchain_traces/test_langchain_traces.py index dfe47de3f62..119ec9b3d0e 100644 --- a/src/promptflow/tests/test_configs/flows/flow_with_langchain_traces/test_langchain_traces.py +++ b/src/promptflow/tests/test_configs/flows/flow_with_langchain_traces/test_langchain_traces.py @@ -1,13 +1,11 @@ import os - from langchain.chat_models import AzureChatOpenAI from langchain_core.messages import HumanMessage from langchain.agents.agent_types import AgentType from langchain.agents.initialize import initialize_agent from langchain.agents.load_tools import load_tools - from promptflow import tool from promptflow.connections import AzureOpenAIConnection from promptflow.integrations.langchain import PromptFlowCallbackHandler @@ -34,5 +32,7 @@ def test_langchain_traces(question: str, conn: AzureOpenAIConnection): content=question ) - return agent.run(message) - + try: + return agent.run(message) + except Exception as e: + return str(e)