agent-api-server 2.1.13__tar.gz → 2.1.14__tar.gz

{agent_api_server-2.1.13 → agent_api_server-2.1.14}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: agent-api-server
-Version: 2.1.13
+Version: 2.1.14
 Summary: A Langgraph agent API server that implements Langgraph agent's web capabilities and can interact with chatbot
 Author: zijie.zhang@advantech.com.cn
 Requires-Python: >=3.10,<3.13
@@ -8,18 +8,14 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Provides-Extra: a2a
 Requires-Dist: a2a-sdk[http-server] (==1.1.2)
-Requires-Dist: a2a-sdk[http-server] (==1.1.2) ; extra == "a2a"
 Requires-Dist: agent-client-protocol (>=0.11.0,<1.0.0)
-Requires-Dist: agent-client-protocol (>=0.11.0,<1.0.0) ; extra == "a2a"
 Requires-Dist: aiofiles (>=24.1.0,<25.0.0)
 Requires-Dist: authlib (>=1.6.5)
 Requires-Dist: cryptography (>=45.0.4,<46.0.0)
 Requires-Dist: fastapi (>=0.117.0,<0.121.2)
 Requires-Dist: fastmcp (>=2.13.0,<3.0.0)
 Requires-Dist: httpx (>=0.27.0,<1.0.0)
-Requires-Dist: httpx (>=0.27.0,<1.0.0) ; extra == "a2a"
 Requires-Dist: langchain-community (==0.3.29)
 Requires-Dist: langchain-core (==0.3.76)
 Requires-Dist: langchain-openai (==0.3.33)
@@ -36,6 +32,7 @@ Requires-Dist: nats-py (>=2.11.0,<3.0.0)
 Requires-Dist: psycopg-binary (>=3.2.9,<4.0.0)
 Requires-Dist: psycopg-pool (>=3.2.6,<4.0.0)
 Requires-Dist: pydantic-settings (>=2.9.1,<3.0.0)
+Requires-Dist: pytest (>=9.1.1,<10.0.0)
 Requires-Dist: redis (>=6.2.0,<7.0.0)
 Requires-Dist: starlette (>=0.49.3,<0.50.0)
 Requires-Dist: tenacity (>=9.1.2,<10.0.0)
@@ -616,13 +613,22 @@ ACP 采用**代理模式（Proxy Mode）**，CLI 进程作为编辑器与 FastAP
 
 ```bash
 # 从 PyPI 安装
-pip install agent-api-server
+pip install agent-api-server==2.1.14
 
 # 或从本地 wheel 安装
-pip install dist/agent_api_server-2.1.9-py3-none-any.whl
+pip install dist/agent_api_server-2.1.14-py3-none-any.whl
 ```
 
-安装后 `agenthub-acp` 命令自动注册到 PATH。
+`2.1.14` 使用本仓库的 LangGraph 0.6 依赖，并包含 WisePortal 组织适配。PyPI 上另有使用 LangGraph 1.x 的 `2.2.x` 版本，安装本版本时请指定版本号。
+
+ACP 命令使用仓库中的 Go 实现，需要在源码目录单独构建，并将生成的程序放入 `PATH`：
+
+```bash
+make -C agent_api_server/acp_bridge/go build
+mkdir -p ~/.local/bin
+install -m 755 agent_api_server/acp_bridge/go/agenthub-acp ~/.local/bin/agenthub-acp
+export PATH="$HOME/.local/bin:$PATH"
+```
 
 **2. 启动 AgentHub 服务**
 
@@ -965,6 +971,73 @@ AGENT_AUTO_REGISTRATION=false
 ENABLE_A2A=false
 ```
 
+### WisePortal 配置
+
+服务启动时会从 `${CONFIG_CENTER_ADDRESS}/v1/public/configs`
+读取 `enableSrpPortal`，保存为 `global_config.isWisePortal`。布尔值 `true` 和
+字符串 `"true"`（忽略大小写和首尾空格）均开启该模式，其余值关闭。
+`CONFIG_CENTER_ADDRESS` 默认是 `http://172.21.92.244:10086`，可通过环境变量覆盖。
+配置读取失败时记录警告并保留当前开关值，初始默认值为 `false`。
+REST 与独立 MCP 服务都会在启动时读取配置；修改配置后需重启服务。
+
+- 关闭时，REST 继续读取 Cookie `TSTenant`，MCP 继续读取 Header `TSTenant`。
+- 开启时，优先读取 Header `Org-Str-Id`，其次 Header `orgStrId`、Cookie `orgStrId`，
+  将该值作为 `TSTenant`。缺少组织字段时不会回退到旧租户字段。
+- A2A 接收同样的 HTTP 组织字段，也支持 `metadata.orgStrId`。
+- SDK 的 MCP 认证头、A2A、Model Manager、SSO 和配置中心客户端会在当前组织上下文中
+  添加 `Org-Str-Id`。组织值不存入共享客户端默认 Header。
+- Go ACP 可通过 `--org-str-id` 或 `ORG_STR_ID` 显式传入组织，代理到后端时携带组织 Header/Cookie。
+
+业务 Agent 自行创建 MCP/HTTP 客户端时，使用共享函数构建每次调用的 Header：
+
+```python
+from agent_api_server.shared.request_context import outbound_headers
+
+headers = outbound_headers({"Authorization": "Bearer ..."}, config=config)
+# 开启 WisePortal 时：{"Authorization": "Bearer ...", "Org-Str-Id": "组织ID"}
+```
+
+`MCPAuthManager.get_headers(config)` 和 `get_headers_safe(config)` 也支持显式传入
+LangGraph 配置；在 SDK 执行的请求中，省略参数会自动使用当前请求的组织。
+业务代码应按请求构建 Header，避免跨组织复用已生成的 Header 字典。
+
+#### WisePortal 模型通知和缓存
+
+配置 `MODEL_MANAGER_NATS_URL` 并开启 WisePortal 后，每个 Agent 接收两类通知：
+
+| Subject | 处理方式 |
+| --- | --- |
+| `model-management-service.model.agent.{agent_name}.{org_str_id}` | 更新或删除当前 Agent 对应工具的模型配置 |
+| `model-management-service.model.tenant.{org_str_id}` | 重新查询当前 Agent 的模型绑定，替换本地快照，并清空已删除的绑定 |
+
+`org_str_id` 支持放在消息顶层或 `message` 内；存在时必须与 subject 的末尾一致。
+没有 `org_str_id` 的旧消息使用 subject 末尾作为标识。`org_id` 仅作为消息元数据保留，
+不会替代 `org_str_id`。模型回调仍使用 `ts_id` 参数，在 WisePortal 模式下该值为 `org_str_id`。
+租户级通知中的通用凭证不会直接写入 Agent 缓存：SDK 会携带 `Org-Str-Id` 查询该 Agent
+自己的模型配置，保存对应 Agent 注册 ID，用于解密模型凭证。
+
+WisePortal 缓存路径为 `data/wiseportal_models/{org_str_id}/{agent_name}.env`，
+其中配置键仍沿用 `CHAT_MODEL_{org_str_id}`、`CHAT_CREDENTIALS_{org_str_id}` 等格式。
+缓存按组织和 Agent 隔离，写入采用文件锁和原子替换；不会将组织模型凭证写入共享进程环境。
+`DynamicLLM` 和模型信息查询会读取最新缓存，因此同一个执行配置也能在下次调用时看到更新或删除。
+模型实例缓存同时区分组织和 Agent 注册 ID。
+
+首次切换时保留旧的 `data/.llm_cfg_env_{tenant_id}` 文件，新的缓存由通知或首次模型查询填充，
+无需手动把旧租户 ID 改成组织 ID。关闭 WisePortal 后继续使用原有缓存路径和 Agent subject。
+自行调用缓存接口的业务代码，应在 WisePortal 模式传入图名称：
+
+```python
+get_env(ts_tenant=org_str_id, agent_name=graph_name)
+set_model_config(..., ts_id=org_str_id, agent_name=graph_name)
+```
+
+通知成功持久化后才 ACK；查询或写入失败会 NAK 请求重投，身份不一致或格式错误的消息会记录错误并终止重投。
+两个 subject 对同一组织的处理会串行执行，防止较慢的租户级刷新覆盖后续 Agent 更新。
+消费者配置更新保留已确认的位置；`ack_wait` 为 30 秒，最多投递 5 次。
+
+回归测试：`python -m pytest tests/test_model_notifications.py tests/test_wiseportal.py tests/test_markdown_fixer.py`。
+可设置 `NATS_TEST_URL` 指向独立的 JetStream 测试实例，启用真实订阅、重投和重连测试。
+
 ### 定义 Agent
 
 创建 `langgraph.json` 配置文件：

{agent_api_server-2.1.13 → agent_api_server-2.1.14}/README.md

@@ -572,13 +572,22 @@ ACP 采用**代理模式（Proxy Mode）**，CLI 进程作为编辑器与 FastAP
 
 ```bash
 # 从 PyPI 安装
-pip install agent-api-server
+pip install agent-api-server==2.1.14
 
 # 或从本地 wheel 安装
-pip install dist/agent_api_server-2.1.9-py3-none-any.whl
+pip install dist/agent_api_server-2.1.14-py3-none-any.whl
 ```
 
-安装后 `agenthub-acp` 命令自动注册到 PATH。
+`2.1.14` 使用本仓库的 LangGraph 0.6 依赖，并包含 WisePortal 组织适配。PyPI 上另有使用 LangGraph 1.x 的 `2.2.x` 版本，安装本版本时请指定版本号。
+
+ACP 命令使用仓库中的 Go 实现，需要在源码目录单独构建，并将生成的程序放入 `PATH`：
+
+```bash
+make -C agent_api_server/acp_bridge/go build
+mkdir -p ~/.local/bin
+install -m 755 agent_api_server/acp_bridge/go/agenthub-acp ~/.local/bin/agenthub-acp
+export PATH="$HOME/.local/bin:$PATH"
+```
 
 **2. 启动 AgentHub 服务**
 
@@ -921,6 +930,73 @@ AGENT_AUTO_REGISTRATION=false
 ENABLE_A2A=false
 ```
 
+### WisePortal 配置
+
+服务启动时会从 `${CONFIG_CENTER_ADDRESS}/v1/public/configs`
+读取 `enableSrpPortal`，保存为 `global_config.isWisePortal`。布尔值 `true` 和
+字符串 `"true"`（忽略大小写和首尾空格）均开启该模式，其余值关闭。
+`CONFIG_CENTER_ADDRESS` 默认是 `http://172.21.92.244:10086`，可通过环境变量覆盖。
+配置读取失败时记录警告并保留当前开关值，初始默认值为 `false`。
+REST 与独立 MCP 服务都会在启动时读取配置；修改配置后需重启服务。
+
+- 关闭时，REST 继续读取 Cookie `TSTenant`，MCP 继续读取 Header `TSTenant`。
+- 开启时，优先读取 Header `Org-Str-Id`，其次 Header `orgStrId`、Cookie `orgStrId`，
+  将该值作为 `TSTenant`。缺少组织字段时不会回退到旧租户字段。
+- A2A 接收同样的 HTTP 组织字段，也支持 `metadata.orgStrId`。
+- SDK 的 MCP 认证头、A2A、Model Manager、SSO 和配置中心客户端会在当前组织上下文中
+  添加 `Org-Str-Id`。组织值不存入共享客户端默认 Header。
+- Go ACP 可通过 `--org-str-id` 或 `ORG_STR_ID` 显式传入组织，代理到后端时携带组织 Header/Cookie。
+
+业务 Agent 自行创建 MCP/HTTP 客户端时，使用共享函数构建每次调用的 Header：
+
+```python
+from agent_api_server.shared.request_context import outbound_headers
+
+headers = outbound_headers({"Authorization": "Bearer ..."}, config=config)
+# 开启 WisePortal 时：{"Authorization": "Bearer ...", "Org-Str-Id": "组织ID"}
+```
+
+`MCPAuthManager.get_headers(config)` 和 `get_headers_safe(config)` 也支持显式传入
+LangGraph 配置；在 SDK 执行的请求中，省略参数会自动使用当前请求的组织。
+业务代码应按请求构建 Header，避免跨组织复用已生成的 Header 字典。
+
+#### WisePortal 模型通知和缓存
+
+配置 `MODEL_MANAGER_NATS_URL` 并开启 WisePortal 后，每个 Agent 接收两类通知：
+
+| Subject | 处理方式 |
+| --- | --- |
+| `model-management-service.model.agent.{agent_name}.{org_str_id}` | 更新或删除当前 Agent 对应工具的模型配置 |
+| `model-management-service.model.tenant.{org_str_id}` | 重新查询当前 Agent 的模型绑定，替换本地快照，并清空已删除的绑定 |
+
+`org_str_id` 支持放在消息顶层或 `message` 内；存在时必须与 subject 的末尾一致。
+没有 `org_str_id` 的旧消息使用 subject 末尾作为标识。`org_id` 仅作为消息元数据保留，
+不会替代 `org_str_id`。模型回调仍使用 `ts_id` 参数，在 WisePortal 模式下该值为 `org_str_id`。
+租户级通知中的通用凭证不会直接写入 Agent 缓存：SDK 会携带 `Org-Str-Id` 查询该 Agent
+自己的模型配置，保存对应 Agent 注册 ID，用于解密模型凭证。
+
+WisePortal 缓存路径为 `data/wiseportal_models/{org_str_id}/{agent_name}.env`，
+其中配置键仍沿用 `CHAT_MODEL_{org_str_id}`、`CHAT_CREDENTIALS_{org_str_id}` 等格式。
+缓存按组织和 Agent 隔离，写入采用文件锁和原子替换；不会将组织模型凭证写入共享进程环境。
+`DynamicLLM` 和模型信息查询会读取最新缓存，因此同一个执行配置也能在下次调用时看到更新或删除。
+模型实例缓存同时区分组织和 Agent 注册 ID。
+
+首次切换时保留旧的 `data/.llm_cfg_env_{tenant_id}` 文件，新的缓存由通知或首次模型查询填充，
+无需手动把旧租户 ID 改成组织 ID。关闭 WisePortal 后继续使用原有缓存路径和 Agent subject。
+自行调用缓存接口的业务代码，应在 WisePortal 模式传入图名称：
+
+```python
+get_env(ts_tenant=org_str_id, agent_name=graph_name)
+set_model_config(..., ts_id=org_str_id, agent_name=graph_name)
+```
+
+通知成功持久化后才 ACK；查询或写入失败会 NAK 请求重投，身份不一致或格式错误的消息会记录错误并终止重投。
+两个 subject 对同一组织的处理会串行执行，防止较慢的租户级刷新覆盖后续 Agent 更新。
+消费者配置更新保留已确认的位置；`ack_wait` 为 30 秒，最多投递 5 次。
+
+回归测试：`python -m pytest tests/test_model_notifications.py tests/test_wiseportal.py tests/test_markdown_fixer.py`。
+可设置 `NATS_TEST_URL` 指向独立的 JetStream 测试实例，启用真实订阅、重投和重连测试。
+
 ### 定义 Agent
 
 创建 `langgraph.json` 配置文件：

{agent_api_server-2.1.13 → agent_api_server-2.1.14}/agent_api_server/a2a_bridge/call_a2a_client.py

@@ -6,6 +6,10 @@ from uuid import uuid4
 
 import httpx
 from agent_api_server.shared.decode_token import decode_jwt
+from agent_api_server.configs import global_config
+from agent_api_server.shared.request_context import (
+    add_organization_header, organization_context, organization_id,
+)
 from a2a.client.client import ClientConfig
 from a2a.client.client_factory import create_client
 from a2a.types import (
@@ -124,6 +128,7 @@ class AgentHubClient:
         if self._client is None:
             httpx_client = httpx.AsyncClient(
                 timeout=httpx.Timeout(self._timeout, connect=10.0),
+                event_hooks={"request": [add_organization_header]},
             )
             client_config = ClientConfig(httpx_client=httpx_client)
 
@@ -162,6 +167,29 @@ class AgentHubClient:
         context_id: str | None = None,
         on_stream_chunk: Callable[[StreamResponse], None] | None = None,
         **extra_metadata,
+    ) -> A2AResult:
+        """Send a request with an organization scoped to this call, not the client."""
+        if global_config.isWisePortal:
+            tenant_id = extra_metadata.get("orgStrId") or organization_id() or tenant_id or ""
+            extra_metadata["orgStrId"] = tenant_id
+        with organization_context(tenant_id):
+            return await self._call(
+                raw_question=raw_question, tenant_id=tenant_id, ei_token=ei_token,
+                files=files, task_id=task_id, context_id=context_id,
+                on_stream_chunk=on_stream_chunk, **extra_metadata,
+            )
+
+    async def _call(
+        self,
+        *,
+        raw_question: str,
+        tenant_id: str = "",
+        ei_token: str = "",
+        files: list | None = None,
+        task_id: str | None = None,
+        context_id: str | None = None,
+        on_stream_chunk: Callable[[StreamResponse], None] | None = None,
+        **extra_metadata,
     ) -> A2AResult:
         """Send a message to the remote agent and consume the full event stream.
 

{agent_api_server-2.1.13 → agent_api_server-2.1.14}/agent_api_server/a2a_bridge/langgraph_executor.py

@@ -10,6 +10,8 @@ from a2a.server.events import EventQueue
 from a2a.server.tasks import TaskUpdater
 from a2a.types import Part, Message, Role, Task, TaskState, TaskStatus
 from agent_api_server.services.config_builder import build_configurable, build_langfuse_callbacks
+from agent_api_server.configs import global_config
+from agent_api_server.shared.request_context import organization_context, organization_id
 
 try:
     from langfuse import propagate_attributes
@@ -194,9 +196,10 @@ class LangGraphAgentExecutor(AgentExecutor):
 
         await updater.start_work()
 
-        graph_task = asyncio.create_task(
-            self._run_graph(graph, user_message, metadata, config, updater, is_continuation)
-        )
+        with organization_context(config["configurable"].get("TSTenant")):
+            graph_task = asyncio.create_task(
+                self._run_graph(graph, user_message, metadata, config, updater, is_continuation)
+            )
         self._running_tasks[task_id] = graph_task
 
         try:
@@ -341,6 +344,8 @@ class LangGraphAgentExecutor(AgentExecutor):
     ) -> Dict[str, Any]:
         metadata = metadata or {}
         ts_tenant = metadata.get("tenant_id", "")
+        if global_config.isWisePortal:
+            ts_tenant = organization_id() or metadata.get("orgStrId", "")
 
         # Namespace thread_id by graph_name to isolate checkpoints
         # across multiple agents sharing the same PostgreSQL checkpointer.

{agent_api_server-2.1.13 → agent_api_server-2.1.14}/agent_api_server/acp_bridge/go/agent.go

@@ -28,7 +28,6 @@ func (e *HTTPError) Error() string {
 	return fmt.Sprintf("Service error: %d", e.StatusCode)
 }
 
-
 // SessionState tracks an active ACP session.
 type SessionState struct {
 	SessionID string `json:"session_id"`
@@ -50,6 +49,7 @@ type AgentHubACPAgent struct {
 	serviceURL     string
 	agentName      string
 	tsTenant       string
+	orgStrID       string
 	eiToken        string
 	agentVersion   string
 	agentDesc      string
@@ -61,7 +61,7 @@ type AgentHubACPAgent struct {
 	httpClient       *http.Client
 	sessions         map[string]*SessionState
 	pendingToolCalls map[string]map[string]*pendingToolCall // sessionID -> tcID -> info
-	streamedNodes    map[string]map[string]bool   // sessionID -> set of nodes
+	streamedNodes    map[string]map[string]bool             // sessionID -> set of nodes
 }
 
 // NewAgentHubACPAgent creates a new agent instance.
@@ -81,7 +81,7 @@ func NewAgentHubACPAgent(
 		agentDesc:        agentDesc,
 		userInputField:   userInputField,
 		extraInputs:      extraInputs,
-		streamMode:     streamMode,
+		streamMode:       streamMode,
 		sessions:         make(map[string]*SessionState),
 		pendingToolCalls: make(map[string]map[string]*pendingToolCall),
 		streamedNodes:    make(map[string]map[string]bool),
@@ -95,6 +95,15 @@ func (a *AgentHubACPAgent) getHTTPClient() *http.Client {
 	return a.httpClient
 }
 
+// addOrganizationHeaders forwards an explicitly configured WisePortal identity.
+// The backend's isWisePortal setting determines whether it replaces TSTenant.
+func (a *AgentHubACPAgent) addOrganizationHeaders(req *http.Request) {
+	if a.orgStrID != "" {
+		req.Header.Set("Org-Str-Id", a.orgStrID)
+		req.AddCookie(&http.Cookie{Name: "orgStrId", Value: a.orgStrID})
+	}
+}
+
 func (a *AgentHubACPAgent) doRequest(method, path string, body io.Reader, params url.Values) (*http.Response, error) {
 	u := a.serviceURL + path
 	if params != nil {
@@ -112,6 +121,7 @@ func (a *AgentHubACPAgent) doRequest(method, path string, body io.Reader, params
 	if a.eiToken != "" {
 		req.AddCookie(&http.Cookie{Name: "EIToken", Value: a.eiToken})
 	}
+	a.addOrganizationHeaders(req)
 	return a.httpClient.Do(req)
 }
 
@@ -181,6 +191,7 @@ func (a *AgentHubACPAgent) streamPrompt(sessionID, userText string) error {
 		req.AddCookie(&http.Cookie{Name: "EIToken", Value: a.eiToken})
 	}
 
+	a.addOrganizationHeaders(req)
 	resp, err := a.httpClient.Do(req)
 	if err != nil {
 		return err
@@ -217,7 +228,6 @@ func (a *AgentHubACPAgent) streamPrompt(sessionID, userText string) error {
 	return scanner.Err()
 }
 
-
 // --- Non-streaming (blocking run) ---
 
 func (a *AgentHubACPAgent) nonStreamPrompt(sessionID, userText string) error {
@@ -256,6 +266,7 @@ func (a *AgentHubACPAgent) nonStreamPrompt(sessionID, userText string) error {
 		req.AddCookie(&http.Cookie{Name: "EIToken", Value: a.eiToken})
 	}
 
+	a.addOrganizationHeaders(req)
 	resp, err := a.httpClient.Do(req)
 	if err != nil {
 		return err
@@ -268,9 +279,9 @@ func (a *AgentHubACPAgent) nonStreamPrompt(sessionID, userText string) error {
 	}
 
 	var result struct {
-		Status  string `json:"status"`
+		Status   string `json:"status"`
 		ThreadID string `json:"thread_id"`
-		Result  struct {
+		Result   struct {
 			Type        string `json:"type"`
 			ContentType string `json:"content_type"`
 			Content     string `json:"content"`
@@ -457,14 +468,14 @@ func (a *AgentHubACPAgent) processSSEEvent(sessionID, raw string) {
 			}
 			// Send as tool_call (for tool list)
 			a.sendStartToolCall(sessionID, tcID, title, nil, text, "completed")
-			
+
 			// Only send as agent_message_chunk if this node hasn't streamed tokens yet
 			// (to avoid duplicate content in chat area)
 			a.mu.Lock()
 			streamed := a.streamedNodes[sessionID]
 			hasStreamed := streamed != nil && streamed[node]
 			a.mu.Unlock()
-			
+
 			if !hasStreamed {
 				a.sendAgentMessage(sessionID, text)
 			}
@@ -538,7 +549,7 @@ func truncate(s string, maxLen int) string {
 		return s
 	}
 	return s[:maxLen]
-}// sendStartToolCallWithContent sends a tool call with content field (for tools_message).
+} // sendStartToolCallWithContent sends a tool call with content field (for tools_message).
 func (a *AgentHubACPAgent) sendStartToolCallWithContent(sessionID, tcID, title, contentText string) {
 	a.sendNotification(sessionID, "session/update", map[string]interface{}{
 		"sessionId": sessionID,
@@ -558,5 +569,3 @@ func (a *AgentHubACPAgent) sendStartToolCallWithContent(sessionID, tcID, title,
 		},
 	})
 }
-
-

agent_api_server-2.1.14/agent_api_server/acp_bridge/go/agent_test.go

@@ -0,0 +1,46 @@
+package main
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestOrganizationForwarding(t *testing.T) {
+	for _, org := range []string{"", "org-a"} {
+		t.Run("organization="+org, func(t *testing.T) {
+			server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+				if got := r.Header.Get("Org-Str-Id"); got != org {
+					t.Errorf("Org-Str-Id = %q, want %q", got, org)
+				}
+				cookie, err := r.Cookie("TSTenant")
+				if err != nil || cookie.Value != "legacy" {
+					t.Error("legacy tenant cookie was not preserved")
+				}
+				cookie, err = r.Cookie("orgStrId")
+				if org == "" {
+					if err != http.ErrNoCookie {
+						t.Error("unexpected orgStrId cookie in legacy request")
+					}
+				} else if err != nil || cookie.Value != org {
+					t.Error("missing organization cookie")
+				}
+				w.WriteHeader(http.StatusOK)
+			}))
+			defer server.Close()
+			agent := NewAgentHubACPAgent(server.URL, "agent", "legacy", "", "1", "", "user_input", nil, true)
+			agent.orgStrID = org
+			agent.sessions["session"] = &SessionState{ThreadID: "thread"}
+			response, err := agent.doRequest(http.MethodGet, "/probe", nil, nil)
+			if err != nil {
+				t.Fatal(err)
+			}
+			if err := response.Body.Close(); err != nil {
+				t.Fatal(err)
+			}
+			if err := agent.streamPrompt("session", "hello"); err != nil {
+				t.Fatal(err)
+			}
+		})
+	}
+}

{agent_api_server-2.1.13 → agent_api_server-2.1.14}/agent_api_server/acp_bridge/go/main.go

@@ -27,6 +27,7 @@ func main() {
 	serviceURL := flag.String("service-url", envOr("AGENT_SERVICE_URL", "http://localhost:8080"), "Base URL of the AgentHub service")
 	agentName := flag.String("agent-name", envOr("AGENT_NAME", ""), "Agent (graph_name) to proxy")
 	tsTenant := flag.String("ts-tenant", envOr("TS_TENANT", ""), "TSTenant auth cookie")
+	orgStrID := flag.String("org-str-id", envOr("ORG_STR_ID", ""), "WisePortal organization ID")
 	eiToken := flag.String("ei-token", envOr("EI_TOKEN", ""), "EIToken auth cookie")
 	agentVersion := flag.String("agent-version", envOr("AGENT_VERSION", "0.1.0"), "Agent version string")
 	userInputField := flag.String("user-input-field", envOr("USER_INPUT_FIELD", "user_input"), "Field name for user input")
@@ -88,6 +89,7 @@ func main() {
 		*serviceURL, *agentName, *tsTenant, *eiToken,
 		*agentVersion, "", *userInputField, extraInputs, *streamMode,
 	)
+	agent.orgStrID = *orgStrID
 
 	if err := runStdioServer(agent); err != nil {
 		log.Fatalf("Server error: %v", err)
@@ -112,6 +114,8 @@ Flags:
     	(env: AGENT_NAME)
   -ts-tenant string
     	TSTenant auth cookie (env: TS_TENANT)
+  -org-str-id string
+        WisePortal organization ID (env: ORG_STR_ID)
   -ei-token string
     	EIToken auth cookie (env: EI_TOKEN)
   -agent-version string
@@ -300,7 +304,7 @@ func (a *AgentHubACPAgent) handleLoadSession(req jsonrpcRequest) *jsonrpcRespons
 
 func (a *AgentHubACPAgent) loadSessionHistory(sessionID, cwd string) {
 	params := url.Values{
-		"user_input_field":     {a.userInputField},
+		"user_input_field": {a.userInputField},
 	}
 	resp, err := a.doRequest("GET", "/api/v1/thread/"+sessionID+"/history", nil, params)
 	if err != nil {
@@ -315,7 +319,7 @@ func (a *AgentHubACPAgent) loadSessionHistory(sessionID, cwd string) {
 	}
 
 	var data struct {
-		Messages       []map[string]interface{} `json:"messages"`
+		Messages []map[string]interface{} `json:"messages"`
 	}
 	if err := json.NewDecoder(resp.Body).Decode(&data); err != nil {
 		log.Printf("Failed to decode session history: %v", err)
@@ -327,7 +331,6 @@ func (a *AgentHubACPAgent) loadSessionHistory(sessionID, cwd string) {
 	log.Printf("Loaded session %s with %d messages", sessionID, len(data.Messages))
 }
 
-
 // replayMessages replays session history using simple messages list.
 func (a *AgentHubACPAgent) replayMessages(sessionID string, messages []map[string]interface{}) {
 	// Pre-scan: collect tool results by name for including in tool_call events
@@ -714,9 +717,9 @@ func cmdLoadSession(serviceURL, agentName, sessionID string, limit int, jsonOutp
 	}
 
 	var data struct {
-		Messages []map[string]interface{} `json:"messages"`
-		GraphName string                  `json:"graph_name"`
-		Total    int                      `json:"total"`
+		Messages  []map[string]interface{} `json:"messages"`
+		GraphName string                   `json:"graph_name"`
+		Total     int                      `json:"total"`
 	}
 	json.Unmarshal(body, &data)
 

{agent_api_server-2.1.13 → agent_api_server-2.1.14}/agent_api_server/api/v1/thread.py

@@ -24,6 +24,7 @@ from agent_api_server.shared.message import (
 from langgraph.pregel import Pregel
 from agent_api_server.services.config_builder import build_configurable
 from agent_api_server.services.stream_service import extract_user_id, execute_graph_stream
+from agent_api_server.shared.request_context import request_tenant, organization_context
 
 logger = logging.getLogger(__name__)
 api_router = APIRouter()
@@ -239,7 +240,7 @@ async def stream(
     cookie_header = request.headers.get("cookie", "")
     cookie = SimpleCookie()
     cookie.load(cookie_header)
-    ts_tenant = cookie.get("TSTenant").value if "TSTenant" in cookie else None
+    ts_tenant = request_tenant(request)
     ei_token = cookie.get("EIToken").value if "EIToken" in cookie else None
 
     state = await _get_thread_or_404(thread_id, get_graph_name)
@@ -316,7 +317,7 @@ async def run_thread(
     cookie_header = request.headers.get("cookie", "")
     cookie = SimpleCookie()
     cookie.load(cookie_header)
-    ts_tenant = cookie.get("TSTenant").value if "TSTenant" in cookie else None
+    ts_tenant = request_tenant(request)
     ei_token = cookie.get("EIToken").value if "EIToken" in cookie else None
 
     storage = AsyncRedisThreadStorage.get_worker_instance()
@@ -434,13 +435,14 @@ async def _execute_graph_with_redis_check(
             else inputs or {}
         )
 
-        execute_task = asyncio.create_task(
-            graph_instance.ainvoke(
-                invoke_input,
-                config=run_cfg,
-                stream_mode=["values"]
+        with organization_context(run_cfg.get("configurable", {}).get("TSTenant")):
+            execute_task = asyncio.create_task(
+                graph_instance.ainvoke(
+                    invoke_input,
+                    config=run_cfg,
+                    stream_mode=["values"]
+                )
             )
-        )
 
         while not execute_task.done():
             if await storage.should_stop(thread_id):

{agent_api_server-2.1.13 → agent_api_server-2.1.14}/agent_api_server/app.py

@@ -15,6 +15,8 @@ from agent_api_server.cache.redis_cache import AsyncRedisThreadStorage
 from agent_api_server.configs import global_config
 from agent_api_server.memory.postgres import AsyncPostgresCheckpointer
 from agent_api_server.shared.exceptions import AgentHubException
+from agent_api_server.shared.request_context import OrganizationContextMiddleware
+from agent_api_server.config_center.config_center import initialize_wiseportal_config
 
 import logging
 
@@ -72,6 +74,8 @@ async def lifespan(app: FastAPI) -> AsyncIterator[None]:
     cache = AsyncRedisThreadStorage.get_worker_instance()
 
     try:
+        await initialize_wiseportal_config()
+
         # 1. Infrastructure (always run, these are per-worker)
         await checkpointer.initialize()
         await cache.initialize()
@@ -134,6 +138,7 @@ async def _cleanup_resources():
 def create_fastapi_app() -> FastAPI:
     """Create and configure FastAPI application."""
     app = FastAPI(lifespan=lifespan)
+    app.add_middleware(OrganizationContextMiddleware)
     
     # Mount static files
     client_dir = Path(__file__).parent / "client"