Merge pull request #86 from zTaoplus/main

docs: update quick start
tablegpt · Nov 14, 2024 · 41a2283 · 41a2283
2 parents a759825 + 6b427a3
commit 41a2283
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/docs/tutorials/quickstart.md b/docs/tutorials/quickstart.md
@@ -42,9 +42,10 @@ pip install langchain-openai
 >>> from langchain_openai import ChatOpenAI
 >>> from pybox import LocalPyBoxManager
 >>> from tablegpt.agent import create_tablegpt_graph
+>>> from tablegpt import DEFAULT_TABLEGPT_IPYKERNEL_PROFILE_DIR
 
 >>> llm = ChatOpenAI(openai_api_base="YOUR_VLLM_URL", openai_api_key="whatever", model_name="TableGPT2-7B")
->>> pybox_manager = LocalPyBoxManager()
+>>> pybox_manager = LocalPyBoxManager(profile_dir=DEFAULT_TABLEGPT_IPYKERNEL_PROFILE_DIR)
 
 >>> agent = create_tablegpt_graph(
 ...     llm=llm,
@@ -97,14 +98,15 @@ from langchain_core.messages import HumanMessage
 from langchain_openai import ChatOpenAI
 from pybox import LocalPyBoxManager
 from tablegpt.agent import create_tablegpt_graph
+from tablegpt import DEFAULT_TABLEGPT_IPYKERNEL_PROFILE_DIR
 
 
 # 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")
 
     # 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,