mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-06 14:22:47 +02:00
chore: linting
This commit is contained in:
parent
489dd0aa52
commit
4e174f17f2
8 changed files with 15 additions and 16 deletions
|
|
@ -130,7 +130,9 @@ async def create_multi_agent_chat_deep_agent(
|
|||
|
||||
_t0 = time.perf_counter()
|
||||
try:
|
||||
mcp_tools_by_agent = await load_mcp_tools_by_connector(db_session, search_space_id)
|
||||
mcp_tools_by_agent = await load_mcp_tools_by_connector(
|
||||
db_session, search_space_id
|
||||
)
|
||||
except Exception as e:
|
||||
# Degrade to builtins-only rather than aborting the turn: a transient
|
||||
# DB or MCP-server hiccup should not deny the user a response.
|
||||
|
|
|
|||
|
|
@ -9,4 +9,6 @@ from ..shared.flags import enabled
|
|||
|
||||
|
||||
def build_doom_loop_mw(flags: AgentFeatureFlags) -> DoomLoopMiddleware | None:
|
||||
return DoomLoopMiddleware(threshold=3) if enabled(flags, "enable_doom_loop") else None
|
||||
return (
|
||||
DoomLoopMiddleware(threshold=3) if enabled(flags, "enable_doom_loop") else None
|
||||
)
|
||||
|
|
|
|||
|
|
@ -78,9 +78,7 @@ def build_permission_context(
|
|||
Rule(permission=tool_def.name, pattern="*", action="deny")
|
||||
)
|
||||
if synthesized:
|
||||
rulesets.append(
|
||||
Ruleset(rules=synthesized, origin="connector_synthesized")
|
||||
)
|
||||
rulesets.append(Ruleset(rules=synthesized, origin="connector_synthesized"))
|
||||
|
||||
general_purpose_interrupt_on: dict[str, bool] = {
|
||||
rule.permission: True
|
||||
|
|
|
|||
|
|
@ -29,7 +29,6 @@ class RateLimitError(Exception):
|
|||
|
||||
|
||||
class _AlwaysFailingChatModel(BaseChatModel):
|
||||
|
||||
@property
|
||||
def _llm_type(self) -> str:
|
||||
return "always-failing-test-model"
|
||||
|
|
|
|||
|
|
@ -67,9 +67,7 @@ class _RecordingChatModel(BaseChatModel):
|
|||
) -> ChatResult:
|
||||
self.call_count += 1
|
||||
return ChatResult(
|
||||
generations=[
|
||||
ChatGeneration(message=AIMessage(content=self.response_text))
|
||||
]
|
||||
generations=[ChatGeneration(message=AIMessage(content=self.response_text))]
|
||||
)
|
||||
|
||||
async def _agenerate(
|
||||
|
|
|
|||
|
|
@ -47,7 +47,9 @@ def test_extract_text_content_ignores_thinking_blocks_and_keeps_markdown_text()
|
|||
assert extract_text_content(content).strip() == markdown.strip()
|
||||
|
||||
|
||||
def test_extract_text_content_returns_empty_when_only_thinking_blocks_are_present() -> None:
|
||||
def test_extract_text_content_returns_empty_when_only_thinking_blocks_are_present() -> (
|
||||
None
|
||||
):
|
||||
content = [
|
||||
{"type": "thinking", "thinking": "No durable fact."},
|
||||
{"type": "thinking", "thinking": "Return no update."},
|
||||
|
|
|
|||
|
|
@ -141,9 +141,7 @@ class TestNormalizeDecision:
|
|||
assert _normalize_permission_decision(decision) == {"decision_type": "reject"}
|
||||
|
||||
def test_lc_envelope_reject_with_message_carries_feedback(self) -> None:
|
||||
decision = {
|
||||
"decisions": [{"type": "reject", "message": "wrong recipient"}]
|
||||
}
|
||||
decision = {"decisions": [{"type": "reject", "message": "wrong recipient"}]}
|
||||
out = _normalize_permission_decision(decision)
|
||||
assert out == {"decision_type": "reject", "feedback": "wrong recipient"}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue