blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source 
Issue from:
https://discord.com/channels/1038097195422978059/1069478035918688346/1112445980466483222
This commit is contained in:
Zander Chase 2023-05-30 01:42:53 -04:00 committed by GitHub
parent 760632b292
commit 26ff18575c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
langchain/callbacks/tracers/langchain_v1.py
View File

@ -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]: