fix ainvoke for agent_graph

code_editing/agents/agent_graph.py

@@ -51,6 +51,11 @@ def invoke(self, *args, **kwargs):
             self.__cached_runnable = self._runnable
         return self.__cached_runnable.invoke(*args, **kwargs)
 
+    async def ainvoke(self, *args, **kwargs):
+        if self.__cached_runnable is None or not self.do_cache:
+            self.__cached_runnable = self._runnable
+        return await self.__cached_runnable.ainvoke(*args, **kwargs)
+
     @property
     def root_params(self):
         return {"tools": self.tools, "llm": self.llm, "run_manager": self.run_manager, "do_cache": self.do_cache}

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ai-agents-code-editing"
-version = "0.1.6"
+version = "0.1.7"
 description = "AI agents for Code Editing"
 authors = ["Alexander Kovrigin <[email protected]>"]
 readme = "README.md"