gomyck-tools 1.3.9__py3-none-any.whl → 1.4.1__py3-none-any.whl

ctools/ai/env_config.py

@@ -5,7 +5,7 @@ __date__ = '2025/5/16 16:42'
 
 import json
 import os
-from typing import Any
+from typing import Any, Optional
 
 from dotenv.main import DotEnv
 
@@ -17,12 +17,23 @@ class Configuration:
     """Initialize configuration with environment variables."""
     if not os.path.exists(dotenv_path): raise FileNotFoundError(f"Could not find .env file at {dotenv_path}")
     self.env = DotEnv(dotenv_path=dotenv_path)
+    for key, value in self.env.dict().items():
+      if value: os.environ[key] = value
 
-  def get_env(self, key: str) -> str:
+  def get_env(self, key: str, default: Optional[Any] = None) -> Any:
     value = self.env.get(key)
-    if value and value.lower() == "true": return True
-    if value and value.lower() == "false": return False
-    return value if value else os.getenv(key)
+    if value:
+      val = value.strip().lower()
+      if val == "true": return True
+      if val == "false": return False
+      return value
+    value = os.getenv(key)
+    if value:
+      val = value.strip().lower()
+      if val == "true": return True
+      if val == "false": return False
+      return value
+    return default
 
   def get_llm_api_key(self) -> str:
     api_key = self.get_env("LLM_API_KEY")

ctools/ai/llm_client.py

@@ -3,7 +3,7 @@ import os
 
 import httpx
 
-from ctools import sys_log, cjson
+from ctools import sys_log, cjson, call
 from ctools.ai.llm_exception import LLMException
 
 logging.getLogger("httpcore").setLevel(logging.WARNING)
@@ -20,11 +20,17 @@ def process_SSE(line):
       return "[DONE]"
     return data
 
-shared_client = httpx.AsyncClient(
-  base_url=os.getenv("LLM_BASE_URL", "https://api.siliconflow.cn/v1/"),
-  timeout=httpx.Timeout(connect=10.0, read=60.0, write=10.0, pool=5.0),
-  limits=httpx.Limits(max_connections=100, max_keepalive_connections=20),
-)
+shared_client = None
+
+@call.once
+def init_shared_client():
+  global shared_client
+  shared_client = httpx.AsyncClient(
+    base_url=os.getenv("LLM_BASE_URL", "https://api.siliconflow.cn/v1/"),
+    timeout=httpx.Timeout(connect=10.0, read=60.0, write=10.0, pool=5.0),
+    limits=httpx.Limits(max_connections=100, max_keepalive_connections=20),
+  )
+  return shared_client
 
 class LLMClient:
   """Manages communication with the LLM provider."""
@@ -51,6 +57,7 @@ class LLMClient:
     self.top_p = top_p
     self.top_k = top_k
     self.frequency_penalty = frequency_penalty
+    init_shared_client()
 
   async def model_completion(self, messages: list[dict[str, str]]):
     self.no_think_compatible(messages)

ctools/ai/mcp/mcp_client.py

@@ -292,6 +292,7 @@ def is_image_content(content: dict) -> bool:
 @asynccontextmanager
 async def init_mcp_clients(mcp_server_config: dict[str, Any]) -> list[MCPClient]:
   mcp_clients = []
+  if not mcp_server_config["mcpServers"]: yield mcp_clients
   for name, sc in mcp_server_config["mcpServers"].items():
     try:
       mc = MCPClient(name, sc)

ctools/ai/tools/quick_tools.py

@@ -5,6 +5,7 @@ __date__ = '2025/6/9 09:49'
 
 import asyncio
 import json
+import os
 import sys
 import uuid
 
@@ -38,10 +39,10 @@ def build_call_back(debug=False):
   call_id = uuid.uuid4()
   queue = asyncio.Queue()
   async def on_msg(cid, role, msg):
-    if debug: print(msg, file=sys.__stdout__, end='')
+    if debug: print(msg, file=sys.__stdout__, end='', flush=True)
     await queue.put({"id": cid, "role": role, "msg": msg})
   async def on_final(cid, is_final, msg):
-    if debug: print(cid, is_final, msg, file=sys.__stdout__)
+    if debug: print(cid, is_final, msg, file=sys.__stdout__, flush=True)
     if is_final:
       await queue.put("[DONE]")
     else:

{gomyck_tools-1.3.9.dist-info → gomyck_tools-1.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gomyck-tools
-Version: 1.3.9
+Version: 1.4.1
 Summary: A tools collection for python development by hao474798383
 Author-email: gomyck <hao474798383@163.com>
 License-Expression: Apache-2.0

{gomyck_tools-1.3.9.dist-info → gomyck_tools-1.4.1.dist-info}/RECORD

@@ -14,15 +14,15 @@ ctools/similar.py,sha256=7mBbp7JrGGyAgA_hDmCvGJ6hGE1Lh43ocyqPkLGclyc,695
 ctools/sys_info.py,sha256=T5pcRblNHEmXEvPk2PPvQga7Hh5aDnNN72pt4U176SE,4270
 ctools/sys_log.py,sha256=T-tgOcrFggSJ2I6pLHhKpj4fvVFk_vbZBOrInKC6y3Q,2789
 ctools/ai/__init__.py,sha256=gTYAICILq48icnFbg0HCbsQO8PbU02EDOQ0JeMvfqTY,98
-ctools/ai/env_config.py,sha256=CQtvQz1f3lvVuDTcnvDDZitnUeUdM3Ta8fXMn-x5Etg,1100
+ctools/ai/env_config.py,sha256=pZ8dG-4-CeMjBdgFHvljY2xMfyBE_0bM961UjIzm6xs,1400
 ctools/ai/llm_chat.py,sha256=dAFar0A5h3ws_kjx9MlWZAawLPYHIwvfdyO_umJgkpU,8655
-ctools/ai/llm_client.py,sha256=iI8kUPn8z0pgTq8mTwnIJm9bOREmSZU5btePinmgGpg,5515
+ctools/ai/llm_client.py,sha256=nQKjQj8WzDqCnMsYag3Td5hYTXep2PthRm6KS7Qyc8U,5661
 ctools/ai/llm_exception.py,sha256=wsCVl0m53Mk7Xfug1obocAthlX0oEo4dytg1eOhWHPg,389
 ctools/ai/mcp/__init__.py,sha256=gTYAICILq48icnFbg0HCbsQO8PbU02EDOQ0JeMvfqTY,98
-ctools/ai/mcp/mcp_client.py,sha256=eDxiG6ZHa4HX-Sz63ObmxyXIq2WZwUxQt8fSw9COfGc,11589
+ctools/ai/mcp/mcp_client.py,sha256=9T22s8BmerT7FrmRJr2VRhvNaV17Xz-C8jGNp08PYu0,11649
 ctools/ai/tools/__init__.py,sha256=gPc-ViRgtFlfX7JUbk5wQZ3wkJ5Ylh14CIqPwa83VPs,98
 ctools/ai/tools/json_extract.py,sha256=bgubZ2RwTo_R1X0CzMnvBWu61hglB1l6oO553645RXc,4842
-ctools/ai/tools/quick_tools.py,sha256=_bJNH4miLltoFjolvJKnOzrArF0hRrXrkBO4elV-Mug,2335
+ctools/ai/tools/quick_tools.py,sha256=WeUFDLLHrP6PCs1pM3G_hRS2YSTp5XjCrJ077u06jeo,2369
 ctools/ai/tools/think_process.py,sha256=RGU9j3_O328Byw05ILek-aMfFBczbly2RA-QRouqUjM,257
 ctools/ai/tools/tool_use_xml_parse.py,sha256=zpHKMhU5LFSpuznU_Z5w_HHp-Bney-Te8DxoU1BNHZ0,1429
 ctools/ai/tools/xml_extract.py,sha256=lpUTERzUQhbiWanc5_ukBP0wPIVg0F9va39S36yyI-4,231
@@ -75,8 +75,8 @@ ctools/web/ctoken.py,sha256=CdHm6-ykBLh7Lv8ZRMunSW40qMTkRH0ITeMLuG9z1ts,883
 ctools/web/download_util.py,sha256=EVN3fxwbUE5Q8nurAya5IYialIKBMoFtSOy0uPKyV-E,1951
 ctools/web/params_util.py,sha256=2vAuwn8OPydv_7BPKypqk4zuerPZqOGcf95wV19H1Ks,824
 ctools/web/upload_util.py,sha256=xH4Z-sz-s7EVDEz2gKXoP9OZTLc-er5yvsiLqhSDFm8,1076
-gomyck_tools-1.3.9.dist-info/licenses/LICENSE,sha256=X25ypfH9E6VTht2hcO8k7LCSdHUcoG_ALQt80jdYZfY,547
-gomyck_tools-1.3.9.dist-info/METADATA,sha256=Z8IRmbiJXNtXNq1ezTsNFHTeLCI1he23ONzg4KOjwY4,1595
-gomyck_tools-1.3.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gomyck_tools-1.3.9.dist-info/top_level.txt,sha256=-MiIH9FYRVKp1i5_SVRkaI-71WmF1sZSRrNWFU9ls3s,7
-gomyck_tools-1.3.9.dist-info/RECORD,,
+gomyck_tools-1.4.1.dist-info/licenses/LICENSE,sha256=X25ypfH9E6VTht2hcO8k7LCSdHUcoG_ALQt80jdYZfY,547
+gomyck_tools-1.4.1.dist-info/METADATA,sha256=CDdnfnwdeNZsLWRZqCrx7cneqAbQgwXraoYDbI4ahg0,1595
+gomyck_tools-1.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+gomyck_tools-1.4.1.dist-info/top_level.txt,sha256=-MiIH9FYRVKp1i5_SVRkaI-71WmF1sZSRrNWFU9ls3s,7
+gomyck_tools-1.4.1.dist-info/RECORD,,