oocana 0.16.5__tar.gz → 0.16.7__tar.gz

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

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

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

@@ -136,7 +136,7 @@ class Context:
             "api_key": os.getenv("OOMOL_LLM_API_KEY", ""),
             "models": os.getenv("OOMOL_LLM_MODELS", "").split(","),
         }
-    
+
     @property
     def host_info(self) -> HostInfo:
         """this is a dict contains the host information
@@ -145,7 +145,17 @@ class Context:
             "gpu_vendor": os.getenv("OOMOL_HOST_GPU_VENDOR", "unknown"),
             "gpu_renderer": os.getenv("OOMOL_HOST_GPU_RENDERER", "unknown"),
         }
-    
+
+    @property
+    def host_endpoint(self) -> str | None:
+        """A host endpoint that allows containers to access services running on the host system.
+        
+        Returns:
+            str: The host endpoint if available.
+            None: If the application is running in a cloud environment where no host endpoint is defined.
+        """
+        return os.getenv("OO_HOST_ENDPOINT", None)
+
     @property
     def is_done(self) -> bool:
         return self.__is_done
@@ -255,6 +265,11 @@ class Context:
         )
     
     def __dataframe(self, payload: PreviewPayload) -> PreviewPayload:
+        def not_default_index(df: DataFrame) -> bool:
+            index = df.index.tolist()
+            default_index = list(range(len(df)))
+            return index != default_index
+
         if isinstance(payload, DataFrame):
             payload = { "type": "table", "data": payload }
 
@@ -266,12 +281,28 @@ class Context:
                     data = df.to_dict(orient='split')
                     columns = data.get("columns", [])
                     rows = data.get("data", [])
+                    if not_default_index(df):
+                        index = df.index.tolist()
+                        rows = [[index[i], *rows[i]] for i in range(len(rows))]
+                        columns = ["", *columns]
+
                 elif isinstance(df, PartialDataFrame):
-                    data_columns = loads(df.head(5).to_json(orient='split'))
-                    columns = data_columns.get("columns", [])
-                    rows_head = data_columns.get("data", [])
-                    data_tail = loads(df.tail(5).to_json(orient='split'))
-                    rows_tail = data_tail.get("data", [])
+                    need_add_index = not_default_index(df)
+                    head_data = df.head(5).to_dict(orient='split')
+                    columns = head_data.get("columns", [])
+
+                    rows_head = head_data.get("data", [])
+                    if need_add_index:
+                        columns = ["", *columns]
+                        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')
+                    rows_tail = tail_data.get("data", [])
+                    if need_add_index:
+                        tail_index = tail_data.get("index", [])
+                        rows_tail = [[tail_index[i], *rows_tail[i]] for i in range(len(rows_tail))]
+
                     rows_dots = [["..."] * len(columns)]
                     rows = rows_head + rows_dots + rows_tail
                 else:

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

@@ -3,20 +3,26 @@ from typing import Any, TypedDict, List, Literal, TypeAlias, Union, Protocol, ru
 
 __all__ = ["PreviewPayload", "TablePreviewPayload", "TextPreviewPayload", "JSONPreviewPayload", "ImagePreviewPayload", "MediaPreviewPayload", "PandasPreviewPayload", "DefaultPreviewPayload"]
 
+@runtime_checkable
+class DataFrameIndex(Protocol):
+    def tolist(self) -> Any:
+        ...
+
 # this class is for pandas.DataFrame
 @runtime_checkable
 class DataFrame(Protocol):
 
-    def __dataframe__(self, *args: Any, **kwargs: Any) -> Any:
+    def __len__(self) -> int:
         ...
 
-    def to_dict(self, *args: Any, **kwargs: Any) -> Any:
+    def __dataframe__(self, *args: Any, **kwargs: Any) -> Any:
         ...
 
-@runtime_checkable
-class JsonAble(Protocol):
+    @property
+    def index(self) -> DataFrameIndex:
+        ...
 
-    def to_json(self, *args: Any, **kwargs: Any) -> Any:
+    def to_dict(self, orient: Literal["split"]) -> Any:
         ...
 
 @runtime_checkable
@@ -27,11 +33,11 @@ class ShapeDataFrame(DataFrame, Protocol):
         ...
 
 @runtime_checkable
-class PartialDataFrame(Protocol):
-    def head(self, *args: Any, **kwargs: Any) -> JsonAble:
+class PartialDataFrame(DataFrame, Protocol):
+    def head(self, count: int) -> DataFrame:
         ...
     
-    def tail(self, *args: Any, **kwargs: Any) -> JsonAble:
+    def tail(self, count: int) -> DataFrame:
         ...
 
 class TablePreviewData(TypedDict):

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

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