Skip to content
6 changes: 3 additions & 3 deletions src/codeas/core/agent.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from typing import Union
from typing import Optional, Union

from pydantic import BaseModel
from tokencost import (
Expand Down Expand Up @@ -46,7 +46,7 @@ class Agent(BaseModel):
instructions: str
model: str
response_format: object = None
system_prompt: str = None
system_prompt: Optional[str] = None

def run(
self,
Expand Down Expand Up @@ -218,4 +218,4 @@ def _calculate_tokens_and_cost(self, messages: list, response=None):
repo = Repo(repo_path=".")
incl_files = repo.filter_files()
files_paths = [path for path, incl in zip(repo.files_paths, incl_files) if incl]
...
...