oocana 0.16.7__tar.gz → 0.16.9__tar.gz

{oocana-0.16.7 → oocana-0.16.9}/PKG-INFO

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

{oocana-0.16.7 → oocana-0.16.9}/oocana/__init__.py

@@ -3,5 +3,6 @@ from .context import *  # noqa: F403
 from .service import *  # noqa: F403
 from .handle_data import *  # noqa: F403
 from .preview import *  # noqa: F403
+from .extra import *  # noqa: F403
 from .schema import *  # noqa: F403
-from .mainframe import Mainframe as Mainframe  # noqa: F403
+from .mainframe import Mainframe as Mainframe  # noqa: F403

{oocana-0.16.7 → oocana-0.16.9}/oocana/context.py

@@ -278,7 +278,7 @@ class Context:
             if isinstance(df, ShapeDataFrame):
                 row_count = df.shape[0]
                 if row_count <= 10:
-                    data = df.to_dict(orient='split')
+                    data = loads(df.to_json(orient='split'))
                     columns = data.get("columns", [])
                     rows = data.get("data", [])
                     if not_default_index(df):
@@ -287,8 +287,9 @@ class Context:
                         columns = ["", *columns]
 
                 elif isinstance(df, PartialDataFrame):
-                    need_add_index = not_default_index(df)
-                    head_data = df.head(5).to_dict(orient='split')
+                    need_add_index = False # TODO: some index is not begin with 0, current just always hide index
+                    # to_json will serialize some default json dumps not supported type like datetime, so we just use to_json for now.
+                    head_data = loads(df.head(5).to_json(orient='split'))
                     columns = head_data.get("columns", [])
 
                     rows_head = head_data.get("data", [])
@@ -297,7 +298,7 @@ class Context:
                         head_index = head_data.get("index", [])
                         rows_head = [[head_index[i], *rows_head[i]] for i in range(len(rows_head))]
                     
-                    tail_data = df.tail(5).to_dict(orient='split')
+                    tail_data = loads(df.tail(5).to_json(orient='split'))
                     rows_tail = tail_data.get("data", [])
                     if need_add_index:
                         tail_index = tail_data.get("index", [])

oocana-0.16.9/oocana/extra.py

@@ -0,0 +1,13 @@
+from typing import Literal, TypedDict
+
+__all__ = ["LLMModelOptions", "LLMMessage"]
+
+class LLMModelOptions(TypedDict):
+    model: str
+    temperature: float
+    top_p: float
+    max_tokens: int
+
+class LLMMessage(TypedDict):
+    role: Literal["system", "user", "assistant"]
+    content: str

{oocana-0.16.7 → oocana-0.16.9}/oocana/preview.py

@@ -22,6 +22,9 @@ class DataFrame(Protocol):
     def index(self) -> DataFrameIndex:
         ...
 
+    def to_json(self, orient: Literal["split"]) -> str:
+        ...
+
     def to_dict(self, orient: Literal["split"]) -> Any:
         ...
 

{oocana-0.16.7 → oocana-0.16.9}/pyproject.toml

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