add some envs for tools and memory (#247)

Author: yaozheng-fang

veadk/memory/long_term_memory.py

@@ -104,6 +104,15 @@ def model_post_init(self, __context: Any) -> None:
 
         # Once user define backend config, use it directly
         if self.backend_config:
+            if "index" not in self.backend_config:
+                logger.warning(
+                    "Attribute `index` not provided in backend_config, use `index` or `app_name` instead."
+                )
+                self.backend_config["index"] = self.index or self.app_name
+
+            logger.debug(
+                f"Init {self.backend}, Use provided backend config: {self.backend_config}"
+            )
             self._backend = _get_backend_cls(self.backend)(**self.backend_config)
             return
 

veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py

@@ -49,7 +49,23 @@ class VikingDBLTMBackend(BaseLongTermMemoryBackend):
     region: str = "cn-beijing"
     """VikingDB memory region"""
 
+    memory_type: list[str] = Field(default_factory=list)
+
     def model_post_init(self, __context: Any) -> None:
+        # We get memory type from:
+        # 1. user input
+        # 2. environment variable
+        # 3. default value
+        if not self.memory_type:
+            env_memory_type = os.getenv("DATABASE_VIKING_MEMORY_TYPE")
+            if env_memory_type:
+                # "event_1, event_2" -> ["event_1", "event_2"]
+                self.memory_type = [x.strip() for x in env_memory_type.split(",")]
+            else:
+                self.memory_type = ["sys_event_v1", "event_v1"]
+
+        logger.info(f"Using memory type: {self.memory_type}")
+
         # check whether collection exist, if not, create it
         if not self._collection_exist():
             self._create_collection()
@@ -135,10 +151,7 @@ def save_memory(self, user_id: str, event_strings: list[str], **kwargs) -> bool:
     def search_memory(
         self, user_id: str, query: str, top_k: int, **kwargs
     ) -> list[str]:
-        filter = {
-            "user_id": user_id,
-            "memory_type": ["sys_event_v1", "event_v1"],
-        }
+        filter = {"user_id": user_id, "memory_type": self.memory_type}
 
         logger.debug(
             f"Request for search memory in VikingDB: filter={filter}, collection_name={self.index}, query={query}, limit={top_k}"

veadk/tools/builtin_tools/run_code.py

@@ -36,9 +36,16 @@ def run_code(code: str, language: str, tool_context: ToolContext) -> str:
 
     tool_id = getenv("AGENTKIT_TOOL_ID")
     host = getenv("AGENTKIT_TOOL_HOST")  # temporary host for code run tool
+    service = getenv(
+        "AGENTKIT_TOOL_SERVICE_CODE"
+    )  # temporary service for code run tool
+    region = getenv("AGENTKIT_TOOL_REGION", "cn-beijing")
+
     session_id = tool_context._invocation_context.session.id
 
-    logger.debug(f"Running code in language: {language}, session_id={session_id}")
+    logger.debug(
+        f"Running code in language: {language}, session_id={session_id}, code={code}, tool_id={tool_id}, host={host}, service={service}, region={region}"
+    )
 
     access_key = getenv("VOLCENGINE_ACCESS_KEY")
     secret_key = getenv("VOLCENGINE_SECRET_KEY")
@@ -59,9 +66,9 @@ def run_code(code: str, language: str, tool_context: ToolContext) -> str:
         action="InvokeTool",
         ak=access_key,
         sk=secret_key,
-        service="agentkit_stg",
+        service=service,
         version="2025-10-30",
-        region="cn-beijing",
+        region=region,
         host=host,
     )