fix assert used

This commit is contained in:
Michael Ingvarsson
2026-07-16 11:48:55 +02:00
parent a341a67727
commit cf1e33474e
3 changed files with 49 additions and 5 deletions
+4 -2
View File
@@ -71,7 +71,8 @@ class BaseAgent(Agent):
"""Run a single interaction with the agent."""
if self.model is None:
await self.initialize()
assert self.model is not None
if self.model is None:
raise RuntimeError("Model initialization failed: model is None")
messages: list[dict[str, str]] = [
{"role": "system", "content": self.system_prompt},
@@ -91,7 +92,8 @@ class BaseAgent(Agent):
"""Run the agent with tool calling capability."""
if self.model is None:
await self.initialize()
assert self.model is not None
if self.model is None:
raise RuntimeError("Model initialization failed: model is None")
try:
capture = _ActResponseCapture()
+6 -3
View File
@@ -489,7 +489,8 @@ class IssueTaskProcessor(TaskProcessor):
return f"SKIP: Awaiting human reply on issue #{self.item.task_number} or PR."
issue_info = self.item.task_info
assert isinstance(issue_info, IssueModel)
if not isinstance(issue_info, IssueModel):
raise TypeError("Expected task_info to be an IssueModel")
title = issue_info.title
issue_body = issue_info.body or "No description provided."
@@ -738,7 +739,8 @@ class AgentDispatcher:
def _build_pr_mission(self, item: WorkItem) -> str:
pr_info = item.task_info
assert isinstance(pr_info, PullRequestModel)
if not isinstance(pr_info, PullRequestModel):
raise TypeError("Expected task_info to be a PullRequestModel")
processor = PRTaskProcessor(
client=self._client,
tools=self._tools,
@@ -751,7 +753,8 @@ class AgentDispatcher:
def _build_issue_mission(self, item: WorkItem) -> str:
issue_info = item.task_info
assert isinstance(issue_info, IssueModel)
if not isinstance(issue_info, IssueModel):
raise TypeError("Expected task_info to be an IssueModel")
processor = IssueTaskProcessor(
client=self._client,
tools=self._tools,