fix /learn TypeError (jupyterlab#286)

dlqqq · Jul 21, 2023 · 4049084 · 4049084
1 parent e2142fe
commit 4049084
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py
@@ -135,7 +135,7 @@ def _build_list_response(self):
 
     async def learn_dir(self, path: str, chunk_size: int, chunk_overlap: int):
         dask_client = await self.dask_client_future
-        splitter_kwargs = {chunk_size: chunk_size, chunk_overlap: chunk_overlap}
+        splitter_kwargs = {"chunk_size": chunk_size, "chunk_overlap": chunk_overlap}
         splitters = {
             ".py": PythonCodeTextSplitter(**splitter_kwargs),
             ".md": MarkdownTextSplitter(**splitter_kwargs),