format and assgin pyarrow

Author: royzhao

kag/common/tools/search_api/impl/openspg_search_api.py

@@ -7,6 +7,7 @@
 
 logger = logging.getLogger(__name__)
 
+
 @SearchApiABC.register("openspg_search_api", as_default=True)
 class OpenSPGSearchAPI(SearchApiABC):
     def __init__(self, project_id=None, host_addr=None, **kwargs):

kag/solver/pipeline/kag_static_pipeline.py

@@ -131,7 +131,9 @@ async def ainvoke(self, query, **kwargs):
             tasks = await self.planning(query, context, **kwargs)
             if not tasks:
                 think_response = context.kwargs.get("thinker", "")
-                answer = await self.generator.ainvoke(query + "\n" + think_response, context, **kwargs)
+                answer = await self.generator.ainvoke(
+                    query + "\n" + think_response, context, **kwargs
+                )
             else:
                 for task in tasks:
                     context.add_task(task)

requirements.txt

@@ -45,6 +45,7 @@ json_repair
 gitpython
 docstring_parser
 aiolimiter
+pyarrow==19.0.1
 pyodps==0.12.2
 aliyun-log-python-sdk==0.8.8
 pyvis