From 26ff18575cc39b7a8d98257542bb0cf1201391b8 Mon Sep 17 00:00:00 2001 From: Zander Chase <130414180+vowelparrot@users.noreply.github.com> Date: Tue, 30 May 2023 01:42:53 -0400 Subject: [PATCH] Set old LCTracer to default to port 8000 (#5381) Issue from: https://discord.com/channels/1038097195422978059/1069478035918688346/1112445980466483222 --- langchain/callbacks/tracers/langchain_v1.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/langchain/callbacks/tracers/langchain_v1.py b/langchain/callbacks/tracers/langchain_v1.py index 31b7630d..bbd14ff2 100644 --- a/langchain/callbacks/tracers/langchain_v1.py +++ b/langchain/callbacks/tracers/langchain_v1.py @@ -1,12 +1,13 @@ from __future__ import annotations import logging +import os from typing import Any, Optional, Union import requests from langchain.callbacks.tracers.base import BaseTracer -from langchain.callbacks.tracers.langchain import get_endpoint, get_headers +from langchain.callbacks.tracers.langchain import get_headers from langchain.callbacks.tracers.schemas import ( ChainRun, LLMRun, @@ -20,6 +21,10 @@ from langchain.schema import get_buffer_string from langchain.utils import raise_for_status_with_text +def _get_endpoint() -> str: + return os.getenv("LANGCHAIN_ENDPOINT", "http://localhost:8000") + + class LangChainTracerV1(BaseTracer): """An implementation of the SharedTracer that POSTS to the langchain endpoint.""" @@ -27,7 +32,7 @@ class LangChainTracerV1(BaseTracer): """Initialize the LangChain tracer.""" super().__init__(**kwargs) self.session: Optional[TracerSessionV1] = None - self._endpoint = get_endpoint() + self._endpoint = _get_endpoint() self._headers = get_headers() def _convert_to_v1_run(self, run: Run) -> Union[LLMRun, ChainRun, ToolRun]: