mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-17 18:35:19 +02:00
multi_agent_chat: pack_subagent owns per-subagent PermissionMiddleware via Ruleset
This commit is contained in:
parent
67142e68b1
commit
d45dfbfbd6
4 changed files with 38 additions and 27 deletions
|
|
@ -22,6 +22,8 @@ from langchain_core.outputs import ChatGeneration, ChatResult
|
|||
from app.agents.multi_agent_chat.subagents.shared.subagent_builder import (
|
||||
pack_subagent,
|
||||
)
|
||||
from app.agents.new_chat.feature_flags import AgentFeatureFlags
|
||||
from app.agents.new_chat.permissions import Ruleset
|
||||
|
||||
|
||||
class RateLimitError(Exception):
|
||||
|
|
@ -73,14 +75,17 @@ async def test_subagent_recovers_when_primary_llm_fails():
|
|||
responses=[AIMessage(content="recovered via fallback")]
|
||||
)
|
||||
|
||||
spec = pack_subagent(
|
||||
result = pack_subagent(
|
||||
name="resilience_test",
|
||||
description="test subagent",
|
||||
system_prompt="be helpful",
|
||||
tools=[],
|
||||
ruleset=Ruleset(origin="resilience_test", rules=[]),
|
||||
flags=AgentFeatureFlags(),
|
||||
model=primary,
|
||||
middleware_stack={"fallback": ModelFallbackMiddleware(fallback)},
|
||||
)
|
||||
spec = result.spec
|
||||
|
||||
agent = create_agent(
|
||||
model=spec["model"],
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue