sycommon-python-lib 0.1.56b13__tar.gz → 0.1.56b14__tar.gz

{sycommon_python_lib-0.1.56b13 → sycommon_python_lib-0.1.56b14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sycommon-python-lib
-Version: 0.1.56b13
+Version: 0.1.56b14
 Summary: Add your description here
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
@@ -13,6 +13,7 @@ Requires-Dist: kafka-python>=2.3.0
 Requires-Dist: langchain>=1.2.0
 Requires-Dist: langchain-core>=1.2.6
 Requires-Dist: langchain-openai>=1.1.6
+Requires-Dist: langfuse>=3.11.2
 Requires-Dist: langgraph>=1.0.5
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: mysql-connector-python>=9.5.0

{sycommon_python_lib-0.1.56b13 → sycommon_python_lib-0.1.56b14}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sycommon-python-lib"
-version = "0.1.56-beta13"
+version = "0.1.56-beta14"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -14,6 +14,7 @@ dependencies = [
     "langchain>=1.2.0",
     "langchain-core>=1.2.6",
     "langchain-openai>=1.1.6",
+    "langfuse>=3.11.2",
     "langgraph>=1.0.5",
     "loguru>=0.7.3",
     "mysql-connector-python>=9.5.0",

{sycommon_python_lib-0.1.56b13 → sycommon_python_lib-0.1.56b14}/src/sycommon/config/Config.py

@@ -21,6 +21,7 @@ class Config(metaclass=SingletonMeta):
         self.embedding_configs = []
         self.reranker_configs = []
         self.sentry_configs = []
+        self.langfuse_configs = []
         self._process_config()
 
     def get_llm_config(self, model_name):
@@ -47,6 +48,12 @@ class Config(metaclass=SingletonMeta):
                 return sentry
         raise ValueError(f"No configuration found for server: {name}")
 
+    def get_langfuse_config(self, name):
+        for langfuse in self.langfuse_configs:
+            if langfuse.get('name') == name:
+                return langfuse
+        raise ValueError(f"No configuration found for server: {name}")
+
     def _process_config(self):
         llm_config_list = self.config.get('LLMConfig', [])
         for llm_config in llm_config_list:

sycommon_python_lib-0.1.56b14/src/sycommon/config/LangfuseConfig.py

@@ -0,0 +1,15 @@
+from pydantic import BaseModel
+
+
+class LangfuseConfig(BaseModel):
+    name: str
+    secretKey: str
+    publicKey: str
+    baseUrl: str
+    enable: bool
+
+    @classmethod
+    def from_config(cls, server_name: str):
+        from sycommon.config.Config import Config
+        langfuse_config = Config().get_langfuse_config(server_name)
+        return cls(**langfuse_config)

{sycommon_python_lib-0.1.56b13 → sycommon_python_lib-0.1.56b14}/src/sycommon/llm/get_llm.py

@@ -1,4 +1,6 @@
+import os
 from typing import Dict, Type, List, Optional, Callable, Any
+from sycommon.config.Config import Config
 from sycommon.llm.llm_logger import LLMLogger
 from langchain_core.language_models import BaseChatModel
 from langchain_core.runnables import Runnable, RunnableLambda, RunnableConfig
@@ -10,6 +12,7 @@ from pydantic import BaseModel, ValidationError, Field
 from sycommon.config.LLMConfig import LLMConfig
 from sycommon.llm.llm_tokens import TokensCallbackHandler
 from sycommon.logging.kafka_log import SYLogger
+from langfuse.langchain import CallbackHandler
 
 
 class StructuredRunnableWithToken(Runnable):
@@ -230,6 +233,21 @@ def get_llm(
     if not llmConfig:
         raise Exception(f"无效的模型配置：{model}")
 
+    callbacks = [LLMLogger()]
+
+    config = Config().config
+    server_name = config.get('Name', '')
+    langfuse_configs = config.get('LangfuseConfig', [])
+    target_config = next(
+        (item for item in langfuse_configs if item.get('name') == server_name), None)
+    if target_config and target_config.get('enable', False):
+        os.environ["LANGFUSE_SECRET_KEY"] = target_config.get('secretKey', '')
+        os.environ["LANGFUSE_PUBLIC_KEY"] = target_config.get('publicKey', '')
+        os.environ["LANGFUSE_BASE_URL"] = target_config.get('baseUrl', '')
+
+        langfuse_handler = CallbackHandler()
+        callbacks += [langfuse_handler]
+
     llm = init_chat_model(
         model_provider=llmConfig.provider,
         model=llmConfig.model,
@@ -237,7 +255,7 @@ def get_llm(
         api_key="-",
         temperature=0.1,
         streaming=streaming,
-        callbacks=[LLMLogger()]
+        callbacks=callbacks
     )
 
     if llm is None:

{sycommon_python_lib-0.1.56b13 → sycommon_python_lib-0.1.56b14}/src/sycommon_python_lib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sycommon-python-lib
-Version: 0.1.56b13
+Version: 0.1.56b14
 Summary: Add your description here
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
@@ -13,6 +13,7 @@ Requires-Dist: kafka-python>=2.3.0
 Requires-Dist: langchain>=1.2.0
 Requires-Dist: langchain-core>=1.2.6
 Requires-Dist: langchain-openai>=1.1.6
+Requires-Dist: langfuse>=3.11.2
 Requires-Dist: langgraph>=1.0.5
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: mysql-connector-python>=9.5.0

{sycommon_python_lib-0.1.56b13 → sycommon_python_lib-0.1.56b14}/src/sycommon_python_lib.egg-info/SOURCES.txt

@@ -7,6 +7,7 @@ src/sycommon/config/Config.py
 src/sycommon/config/DatabaseConfig.py
 src/sycommon/config/EmbeddingConfig.py
 src/sycommon/config/LLMConfig.py
+src/sycommon/config/LangfuseConfig.py
 src/sycommon/config/MQConfig.py
 src/sycommon/config/RerankerConfig.py
 src/sycommon/config/SentryConfig.py

{sycommon_python_lib-0.1.56b13 → sycommon_python_lib-0.1.56b14}/src/sycommon_python_lib.egg-info/requires.txt

@@ -7,6 +7,7 @@ kafka-python>=2.3.0
 langchain>=1.2.0
 langchain-core>=1.2.6
 langchain-openai>=1.1.6
+langfuse>=3.11.2
 langgraph>=1.0.5
 loguru>=0.7.3
 mysql-connector-python>=9.5.0