feat: tablegpt kernel setups

examples/data_analysis.py

@@ -8,6 +8,7 @@
 from pybox import LocalPyBoxManager
 from tablegpt.agent import create_tablegpt_graph
 from tablegpt.agent.file_reading import Stage
+from tablegpt import DEFAULT_TABLEGPT_IPYKERNEL_PROFILE_DIR
 
 
 class Attachment(TypedDict):
@@ -19,10 +20,16 @@ class Attachment(TypedDict):
 
 # tablegpt-agent fully supports async invocation
 async def main() -> None:
-    llm = ChatOpenAI(openai_api_base="YOUR_VLLM_URL", openai_api_key="whatever", model_name="TableGPT2-7B")
+    llm = ChatOpenAI(
+        openai_api_base="YOUR_VLLM_URL",
+        openai_api_key="whatever",
+        model_name="TableGPT2-7B",
+    )
 
     # Use local pybox manager for development and testing
-    pybox_manager = LocalPyBoxManager()
+    pybox_manager = LocalPyBoxManager(
+        profile_dir=DEFAULT_TABLEGPT_IPYKERNEL_PROFILE_DIR
+    )
 
     agent = create_tablegpt_graph(
         llm=llm,
@@ -37,7 +44,9 @@ async def main() -> None:
     attachment_msg = HumanMessage(
         content="",
         # The dataset can be viewed in examples/datasets/titanic.csv.
-        additional_kwargs={"attachments": [Attachment(filename="titanic.csv")]},
+        additional_kwargs={
+            "attachments": [Attachment(filename="examples/datasets/titanic.csv")]
+        },
     )
     await agent.ainvoke(
         input={

pyproject.toml

@@ -40,6 +40,9 @@ Source = "https://github.com/tablegpt/tablegpt"
 [tool.hatch.build.targets.wheel]
 packages = ["src/tablegpt"]
 
+[tool.hatch.build.targets.wheel.shared-data]
+"ipython-startup-scripts" = "share/ipykernel/profile/tablegpt/startup"
+
 [tool.hatch.version]
 path = "src/tablegpt/__about__.py"
 

src/tablegpt/__init__.py

@@ -0,0 +1,6 @@
+import os
+import sys
+
+DEFAULT_TABLEGPT_IPYKERNEL_PROFILE_DIR = os.path.join(
+    sys.prefix, "share/ipykernel/profile/tablegpt"
+)