Skip to content

Commit 84d10d7

Browse files
committed
fixed mypy findings
1 parent 3c0b70c commit 84d10d7

3 files changed

Lines changed: 8 additions & 5 deletions

File tree

src/config/config_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ def load_config(self) -> OpenRAGConfig:
193193
return self._config
194194

195195
# Start with defaults
196-
config_data = {
196+
config_data: dict[str, Any] = {
197197
"providers": {
198198
"openai": {},
199199
"anthropic": {},

src/models/tasks.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import time
33
from dataclasses import dataclass, field
44
from enum import Enum
5-
from typing import ClassVar, Dict, Optional
5+
from typing import Any, ClassVar, Dict, Optional
66

77

88
class TaskStatus(Enum):
@@ -71,6 +71,8 @@ class UploadTask:
7171
status: TaskStatus = TaskStatus.PENDING
7272
created_at: float = field(default_factory=time.time)
7373
updated_at: float = field(default_factory=time.time)
74+
processor: Optional[Any] = None
75+
background_task: Optional[Any] = None
7476
_sequence_number: int = field(init=False, repr=False)
7577

7678
def __post_init__(self):

src/services/flows_service.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import json
33
import os
44
from datetime import datetime
5+
from typing import Any
56

67
from cachetools import LRUCache
78

@@ -625,7 +626,7 @@ def _get_provider_name_display(self, provider: str):
625626
return "OpenAI"
626627

627628
async def _update_flow_field(
628-
self, flow_id: str, field_name: str, field_value: str, node_display_name: str = None
629+
self, flow_id: str, field_name: str, field_value: Any, node_display_name: str = None
629630
):
630631
"""
631632
Generic helper function to update any field in any Langflow component.
@@ -1032,7 +1033,7 @@ async def change_langflow_model_value(
10321033
# Process all flows simultaneously
10331034
raw_results = await asyncio.gather(*tasks, return_exceptions=True)
10341035

1035-
results = []
1036+
results: list[Any] = []
10361037
for i, result in enumerate(raw_results):
10371038
if isinstance(result, Exception):
10381039
flow_name = flow_configs[i]["name"]
@@ -1237,7 +1238,7 @@ async def wrap_node_update(node, p, m, label):
12371238
"flow_id": flow_id,
12381239
}
12391240

1240-
async def _update_component_langflow(self, template, model: str):
1241+
async def _update_component_langflow(self, template, model: Any):
12411242
# Call custom_component/update endpoint to get updated template
12421243
# Only call if code field exists (custom components should have code)
12431244
if "code" in template and "value" in template["code"]:

0 commit comments

Comments
 (0)