agentic-kit-common 0.0.12__tar.gz → 0.0.14__tar.gz

{agentic_kit_common-0.0.12 → agentic_kit_common-0.0.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-kit-common
-Version: 0.0.12
+Version: 0.0.14
 Summary: Common utilities and tools for agentic kit ecosystem
 Home-page: 
 Author: manson
@@ -24,6 +24,7 @@ Requires-Dist: langchain_core
 Requires-Dist: langgraph
 Requires-Dist: langchain_community
 Requires-Dist: langchain_experimental
+Requires-Dist: langchain-openai
 Requires-Dist: mysql-connector-python
 Requires-Dist: sqlalchemy
 Requires-Dist: sqlglot

agentic_kit_common-0.0.14/agentic_kit_common/llm/__init__.py

@@ -0,0 +1,2 @@
+from .openai import create_openai_llm
+from .utils import combine_simple_context

agentic_kit_common-0.0.14/agentic_kit_common/llm/openai.py

@@ -0,0 +1,33 @@
+import os
+
+from dotenv import load_dotenv
+from langchain_openai import ChatOpenAI
+
+load_dotenv()
+
+
+model_name = os.getenv("MODEL_NAME", None)
+openai_api_base = os.getenv("OPENAI_API_BASE", None)
+openai_api_key = os.getenv("OPENAI_API_KEY", 'API_KEY')
+temperature = float(os.getenv("TEMPERATURE", 0.2))
+
+
+def create_openai_llm(**kwargs):
+    _model_name = kwargs.pop('model_name', model_name)
+    _openai_api_base = kwargs.pop('openai_api_base', openai_api_base)
+    _openai_api_key = kwargs.pop('openai_api_key', openai_api_key)
+    if not _openai_api_key:
+        _openai_api_key = 'API_KEY'
+    _temperature = kwargs.pop('temperature', temperature)
+
+    assert model_name is not None
+    assert openai_api_base is not None
+
+    llm = ChatOpenAI(
+        model_name=_model_name,
+        openai_api_base=_openai_api_base,
+        openai_api_key=_openai_api_key,
+        temperature=_temperature,
+        **kwargs
+    )
+    return llm

agentic_kit_common-0.0.14/agentic_kit_common/llm/utils.py

@@ -0,0 +1,61 @@
+import json
+import re
+from typing import Any, Dict, List, Union
+
+from langchain_core.messages import BaseMessage, SystemMessage, HumanMessage
+
+
+def combine_simple_context(system: str, user: str = None):
+    context: List[Union[BaseMessage, dict[str, Any]]] = [SystemMessage(content=system)]
+    if user:
+        context.append(HumanMessage(content=system))
+    return context
+
+
+def fix_json_response(text: str) -> Union[Dict, List, str, int, float, bool, None]:
+    """
+    去掉大模型返回的 ```json / ``` 等 markdown 标记，并安全反序列化 JSON。
+    若解析失败，返回原字符串。
+
+    参数
+    ----
+    text : str
+        原始响应，可能包含 ```json ... ``` 或其他变体。
+
+    返回
+    ----
+    Python 对象（dict / list / str / int / float / bool / None）
+        解析失败时返回输入字符串本身。
+    """
+    if not isinstance(text, str):
+        return text
+
+    # 1. 去掉 ```json 或 ``` 包裹（支持开头、结尾、单行、多行）
+    cleaned = re.sub(r'^\s*```(?:json|JSON)?\s*\n?', '', text)
+    cleaned = re.sub(r'\n?\s*```\s*$', '', cleaned)
+
+    # 2. 去掉首尾空白
+    cleaned = cleaned.strip()
+
+    # 3. 尝试 JSON 反序列化
+    try:
+        json.loads(cleaned)
+        return cleaned
+    except json.JSONDecodeError:
+        # 4. 兜底：返回原字符串
+        return text
+
+
+# ----------------- 使用示例 -----------------
+if __name__ == "__main__":
+    demo_list = [
+        '```json\n{"a": 1, "b": "hello"}\n```',
+        "```JSON{'key': 'value'}```",
+        "```\n[1, 2, 3]```",
+        "plain text",
+        {"already_dict": 1},
+    ]
+    for d in demo_list:
+        print("原始:", repr(d))
+        print("修复:", repr(fix_json_response(d)))
+        print("-" * 30)

{agentic_kit_common-0.0.12 → agentic_kit_common-0.0.14}/agentic_kit_common/orm/execution.py

@@ -59,7 +59,7 @@ def is_readonly_expression(node: expressions.Expression, allowed_operations: Opt
     if allowed_operations:
         op_type = get_operation_type(node)
         # 检查操作类型
-        if op_type not in allowed_operations:
+        if op_type.lower() not in allowed_operations and op_type.upper() not in allowed_operations:
             return False, op_type
 
     if enable_wildcard_check and isinstance(node, expressions.Select) and node.find(expressions.Star):

{agentic_kit_common-0.0.12 → agentic_kit_common-0.0.14}/agentic_kit_common.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-kit-common
-Version: 0.0.12
+Version: 0.0.14
 Summary: Common utilities and tools for agentic kit ecosystem
 Home-page: 
 Author: manson
@@ -24,6 +24,7 @@ Requires-Dist: langchain_core
 Requires-Dist: langgraph
 Requires-Dist: langchain_community
 Requires-Dist: langchain_experimental
+Requires-Dist: langchain-openai
 Requires-Dist: mysql-connector-python
 Requires-Dist: sqlalchemy
 Requires-Dist: sqlglot

{agentic_kit_common-0.0.12 → agentic_kit_common-0.0.14}/agentic_kit_common.egg-info/SOURCES.txt

@@ -6,6 +6,9 @@ agentic_kit_common.egg-info/SOURCES.txt
 agentic_kit_common.egg-info/dependency_links.txt
 agentic_kit_common.egg-info/requires.txt
 agentic_kit_common.egg-info/top_level.txt
+agentic_kit_common/llm/__init__.py
+agentic_kit_common/llm/openai.py
+agentic_kit_common/llm/utils.py
 agentic_kit_common/log/__init__.py
 agentic_kit_common/log/logger.py
 agentic_kit_common/minio/__init__.py

{agentic_kit_common-0.0.12 → agentic_kit_common-0.0.14}/agentic_kit_common.egg-info/requires.txt

@@ -6,6 +6,7 @@ langchain_core
 langgraph
 langchain_community
 langchain_experimental
+langchain-openai
 mysql-connector-python
 sqlalchemy
 sqlglot

{agentic_kit_common-0.0.12 → agentic_kit_common-0.0.14}/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='agentic-kit-common',
-    version="0.0.12",
+    version="0.0.14",
     author="manson",
     author_email="manson.li3307@gmail.com",
     description='Common utilities and tools for agentic kit ecosystem',
@@ -37,6 +37,7 @@ setup(
         "langgraph",
         "langchain_community",
         "langchain_experimental",
+        "langchain-openai",
 
         # sqlalchemy
         "mysql-connector-python",