Merge branch 'main' of github.com:homanp/superagent

fern/apis/legacy/generators.yml

@@ -1 +1,4 @@
-{}
+api:
+  path: ./openapi/openapi.yaml
+  settings:
+    use-title: false

fern/apis/prod/generators.yml

@@ -1,3 +1,7 @@
+api:
+  path: ./openapi/openapi.yaml
+  settings:
+    use-title: false
 groups:
   publish:
     generators:

fern/fern.config.json

@@ -1,4 +1,4 @@
 {
   "organization": "superagent",
-  "version": "0.16.43"
+  "version": "0.26.3"
 }

libs/superagent/app/agents/base.py

@@ -1,4 +1,5 @@
 from abc import ABC, abstractmethod
+from functools import cached_property
 from typing import Any, List, Optional
 
 from langchain.agents import AgentExecutor
@@ -78,6 +79,11 @@ def prompt(self) -> Any:
     def tools(self) -> Any:
         ...
 
+    # TODO: Set a proper return type when we remove Langchain agent type
+    @cached_property
+    async def memory(self) -> Any:
+        ...
+
     @abstractmethod
     def get_agent(self) -> AgentExecutor:
         ...

libs/superagent/app/agents/langchain.py

@@ -1,4 +1,5 @@
 import datetime
+from functools import cached_property
 
 from decouple import config
 from langchain.agents import AgentType, initialize_agent
@@ -62,9 +63,8 @@ def _get_llm(self):
                 max_tokens=llm_data.params.max_tokens,
             )
 
-    async def _get_memory(
-        self,
-    ) -> None | MotorheadMemory | ConversationBufferWindowMemory:
+    @cached_property
+    async def memory(self) -> None | MotorheadMemory | ConversationBufferWindowMemory:
         # if memory is already set, in the main agent base class, return it
         if not self.session_id:
             raise ValueError("Session ID is required to initialize memory")
@@ -95,7 +95,7 @@ async def _get_memory(
 
     async def get_agent(self):
         llm = self._get_llm()
-        memory = await self._get_memory()
+        memory = await self.memory
         tools = self.tools
         prompt = self.prompt