oocana 0.15.0__tar.gz → 0.15.2__tar.gz

{oocana-0.15.0 → oocana-0.15.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oocana
-Version: 0.15.0
+Version: 0.15.2
 Summary: python implement of oocana to give a context for oocana block
 License: MIT
 Requires-Python: >=3.9

{oocana-0.15.0 → oocana-0.15.2}/oocana/context.py

@@ -20,6 +20,9 @@ class OnlyEqualSelf:
 
 class OOMOL_LLM_ENV(TypedDict):
     base_url: str
+    """{basUrl}/v1 openai compatible endpoint
+    """
+    base_url_v1: str
     api_key: str
     models: list[str]
 
@@ -105,6 +108,7 @@ class Context:
         """
         return {
             "base_url": os.getenv("OOMOL_LLM_BASE_URL", ""),
+            "base_url_v1": os.getenv("OOMOL_LLM_BASE_URL_V1", ""),
             "api_key": os.getenv("OOMOL_LLM_API_KEY", ""),
             "models": os.getenv("OOMOL_LLM_MODELS", "").split(","),
         }

{oocana-0.15.0 → oocana-0.15.2}/oocana/mainframe.py

@@ -10,6 +10,8 @@ from typing import Optional
 
 __all__ = ["Mainframe"]
 
+
+
 class Mainframe:
     address: str
     client: mqtt.Client
@@ -72,12 +74,13 @@ class Mainframe:
     def report(self, block_info: BlockDict, msg: dict) -> mqtt.MQTTMessageInfo:
         return self.client.publish("report", dumps({**block_info, **msg}), qos=1)
     
-    def notify_executor_ready(self, session_id: str, executor_name: str, package: str | None) -> None:
+    def notify_executor_ready(self, session_id: str, executor_name: str, package: str | None, identifier: str | None) -> None:
         self.client.publish(f"session/{session_id}", dumps({
             "type": "ExecutorReady",
             "session_id": session_id,
             "executor_name": executor_name,
             "package": package,
+            "identifier": identifier,
         }), qos=1)
 
     def notify_block_ready(self, session_id: str, job_id: str) -> dict:

{oocana-0.15.0 → oocana-0.15.2}/oocana/throtter.py

@@ -4,8 +4,8 @@ import threading
 def throttle(period):
     last_invoke_time = 0
     timer = None
-    last_args = None
-    last_kwargs = None
+    last_args = []
+    last_kwargs = dict()
 
     def decorator(fn):
         def wrapper(*args, **kwargs):

{oocana-0.15.0 → oocana-0.15.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "oocana"
-version = "0.15.0"
+version = "0.15.2"
 description = "python implement of oocana to give a context for oocana block"
 dependencies = [
     "paho-mqtt>=2",