Set old LCTracer to default to port 8000 (#5381)

Issue from:
https://discord.com/channels/1038097195422978059/1069478035918688346/1112445980466483222
pull/5426/head
Zander Chase 1 year ago committed by GitHub
parent 760632b292
commit 26ff18575c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -1,12 +1,13 @@
from __future__ import annotations from __future__ import annotations
import logging import logging
import os
from typing import Any, Optional, Union from typing import Any, Optional, Union
import requests import requests
from langchain.callbacks.tracers.base import BaseTracer 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 ( from langchain.callbacks.tracers.schemas import (
ChainRun, ChainRun,
LLMRun, LLMRun,
@ -20,6 +21,10 @@ from langchain.schema import get_buffer_string
from langchain.utils import raise_for_status_with_text from langchain.utils import raise_for_status_with_text
def _get_endpoint() -> str:
return os.getenv("LANGCHAIN_ENDPOINT", "http://localhost:8000")
class LangChainTracerV1(BaseTracer): class LangChainTracerV1(BaseTracer):
"""An implementation of the SharedTracer that POSTS to the langchain endpoint.""" """An implementation of the SharedTracer that POSTS to the langchain endpoint."""
@ -27,7 +32,7 @@ class LangChainTracerV1(BaseTracer):
"""Initialize the LangChain tracer.""" """Initialize the LangChain tracer."""
super().__init__(**kwargs) super().__init__(**kwargs)
self.session: Optional[TracerSessionV1] = None self.session: Optional[TracerSessionV1] = None
self._endpoint = get_endpoint() self._endpoint = _get_endpoint()
self._headers = get_headers() self._headers = get_headers()
def _convert_to_v1_run(self, run: Run) -> Union[LLMRun, ChainRun, ToolRun]: def _convert_to_v1_run(self, run: Run) -> Union[LLMRun, ChainRun, ToolRun]:

Loading…
Cancel
Save