github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-18 16:16:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

py tracer fixes (#5377)

Browse Source
This commit is contained in:
Ankush Gola
2023-05-30 18:47:06 -07:00
committed by GitHub
parent ce8b7a2a69
commit 1671c2afb2
5 changed files with 181 additions and 167 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/integration_tests/callbacks/test_langchain_tracer.py
View File

@@ -8,6 +8,7 @@ from aiohttp import ClientSession
from langchain.agents import AgentType, initialize_agent, load_tools
from langchain.callbacks import tracing_enabled
from langchain.callbacks.manager import tracing_v2_enabled
from langchain.chat_models import ChatOpenAI
from langchain.llms import OpenAI
questions = [
@@ -140,10 +141,10 @@ async def test_tracing_v2_environment_variable() -> None:
def test_tracing_v2_context_manager() -> None:
llm = OpenAI(temperature=0)
llm = ChatOpenAI(temperature=0)
tools = load_tools(["llm-math", "serpapi"], llm=llm)
agent = initialize_agent(
tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True
tools, llm, agent=AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION, verbose=True
)
if "LANGCHAIN_TRACING_V2" in os.environ:
del os.environ["LANGCHAIN_TRACING_V2"]