oocana 0.16.7__py3-none-any.whl → 0.16.9__py3-none-any.whl

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/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/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/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.dist-info → oocana-0.16.9.dist-info}/METADATA

@@ -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.9.dist-info/RECORD

@@ -0,0 +1,14 @@
+oocana-0.16.9.dist-info/METADATA,sha256=G_tNXd71n6dTXHfvfkCZkNE6OhGlTDKXNNcXvgzTTS4,221
+oocana-0.16.9.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
+oocana-0.16.9.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+oocana/__init__.py,sha256=kuZa-aY4bqStH_15K4KL9hjOeO9d-M9mJYKs4fm3n08,317
+oocana/context.py,sha256=5wWUZr4BjojizFH-9yeadWvIpiUt2v7wUN3b7y34CFc,14677
+oocana/data.py,sha256=0KpJyckz_X5JbrncBRFbOaU00CxcDx2UYHk6xuH1Poc,2890
+oocana/extra.py,sha256=1f7cu0QA3jnb6PJtEDXMsQOSGWtjUhsRz82DaPRyjKY,289
+oocana/handle_data.py,sha256=p0iEvdsVV3BqcelM8rvq0a_VPI52SeahaGCszfhZ0TI,1927
+oocana/mainframe.py,sha256=Jixk9PjvJh7PnYwPGqOR5viwcOHWfaPhEiQFNHxHCxw,5219
+oocana/preview.py,sha256=qJEoBQWno0BkuU501vd16VwE0okTC3PydFc_jpkwc98,2096
+oocana/schema.py,sha256=8LwMaW4eXa3EmKxR4kzyTOpZiClMRMMsMA6f9CXodW4,4332
+oocana/service.py,sha256=dPCcScQfMBlEjIodFnKU17-HlbBzrQbQK6CNRw7SmOE,2442
+oocana/throtter.py,sha256=xTldm2OZ5uYPiyC6Aatug4FJAoMylPjtXftxHvSr_50,1104
+oocana-0.16.9.dist-info/RECORD,,

oocana-0.16.7.dist-info/RECORD

@@ -1,13 +0,0 @@
-oocana-0.16.7.dist-info/METADATA,sha256=0_TpqDCVlfykzajr-HwTAb9844PjjmtNY9SiL93krig,221
-oocana-0.16.7.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
-oocana-0.16.7.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-oocana/__init__.py,sha256=zMrQ1asZoRE91-BShf1v2ibdThaMc07YtZt5iNbbxsg,281
-oocana/context.py,sha256=LyavX-bgqWGmQkLmE-_QNCd3aHWesShEJJToi4fc32U,14466
-oocana/data.py,sha256=0KpJyckz_X5JbrncBRFbOaU00CxcDx2UYHk6xuH1Poc,2890
-oocana/handle_data.py,sha256=p0iEvdsVV3BqcelM8rvq0a_VPI52SeahaGCszfhZ0TI,1927
-oocana/mainframe.py,sha256=Jixk9PjvJh7PnYwPGqOR5viwcOHWfaPhEiQFNHxHCxw,5219
-oocana/preview.py,sha256=xTyVtf5mKMVlXRwz3JSB-n4e_6dlxalkE341nUDwtD4,2027
-oocana/schema.py,sha256=8LwMaW4eXa3EmKxR4kzyTOpZiClMRMMsMA6f9CXodW4,4332
-oocana/service.py,sha256=dPCcScQfMBlEjIodFnKU17-HlbBzrQbQK6CNRw7SmOE,2442
-oocana/throtter.py,sha256=xTldm2OZ5uYPiyC6Aatug4FJAoMylPjtXftxHvSr_50,1104
-oocana-0.16.7.dist-info/RECORD,,