Skip to content

Commit a350cef

Browse files
committed
add anotations for kwargs
1 parent db123d5 commit a350cef

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

proactiveagent/providers/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
class BaseProvider(ABC):
88
"""Abstract base class for AI providers"""
99

10-
def __init__(self, model: str, **kwargs):
10+
def __init__(self, model: str, **kwargs: Any):
1111
"""
1212
Initialize provider
1313
@@ -24,7 +24,7 @@ async def generate_response(
2424
messages: List[Dict[str, str]],
2525
system_prompt: Optional[str] = None,
2626
triggered_by_user_message: bool = False,
27-
**kwargs
27+
**kwargs: Any
2828
) -> str:
2929
"""
3030
Generate a response from the AI

proactiveagent/providers/openai_provider.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ class ResponseDecision(BaseModel):
2424
class OpenAIProvider(BaseProvider):
2525
"""OpenAI provider for Active AI agents"""
2626

27-
def __init__(self, model: str = "gpt-3.5-turbo", **kwargs):
27+
def __init__(self, model: str = "gpt-3.5-turbo", **kwargs: Any):
2828
"""
2929
Initialize OpenAI provider
3030
@@ -42,7 +42,7 @@ async def generate_response(
4242
messages: List[Dict[str, str]],
4343
system_prompt: Optional[str] = None,
4444
triggered_by_user_message: bool = False,
45-
**kwargs
45+
**kwargs: Any
4646
) -> str:
4747
"""Generate response using OpenAI API"""
4848
def _sync_generate():
@@ -84,7 +84,7 @@ async def should_respond(
8484
messages: List[Dict[str, str]],
8585
elapsed_time: int,
8686
context: Dict[str, Any],
87-
**kwargs
87+
**kwargs: Any
8888
) -> bool:
8989
"""
9090
Determine if AI should respond using OpenAI decision-making with structured output
@@ -155,7 +155,7 @@ async def calculate_sleep_time(
155155
min_sleep_time: int,
156156
max_sleep_time: int,
157157
context: Dict[str, Any],
158-
**kwargs
158+
**kwargs: Any
159159
) -> tuple[int, str]:
160160
"""
161161
Calculate sleep time using OpenAI pattern interpretation with structured output

0 commit comments

Comments
 (0)