aboutsummaryrefslogtreecommitdiff
path: root/stubs/open_webui
diff options
context:
space:
mode:
Diffstat (limited to 'stubs/open_webui')
-rw-r--r--stubs/open_webui/__init__.pyi0
-rw-r--r--stubs/open_webui/models/__init__.pyi0
-rw-r--r--stubs/open_webui/models/config.pyi5
-rw-r--r--stubs/open_webui/models/users.pyi11
-rw-r--r--stubs/open_webui/routers/__init__.pyi0
-rw-r--r--stubs/open_webui/routers/images.pyi14
6 files changed, 30 insertions, 0 deletions
diff --git a/stubs/open_webui/__init__.pyi b/stubs/open_webui/__init__.pyi
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/stubs/open_webui/__init__.pyi
diff --git a/stubs/open_webui/models/__init__.pyi b/stubs/open_webui/models/__init__.pyi
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/stubs/open_webui/models/__init__.pyi
diff --git a/stubs/open_webui/models/config.pyi b/stubs/open_webui/models/config.pyi
new file mode 100644
index 0000000..ae951b2
--- /dev/null
+++ b/stubs/open_webui/models/config.pyi
@@ -0,0 +1,5 @@
+from typing import Any
+
+class Config:
+ @staticmethod
+ async def get(key: str, default: Any = None) -> Any: ...
diff --git a/stubs/open_webui/models/users.pyi b/stubs/open_webui/models/users.pyi
new file mode 100644
index 0000000..79d409e
--- /dev/null
+++ b/stubs/open_webui/models/users.pyi
@@ -0,0 +1,11 @@
+from typing import Any
+
+class UserModel:
+ id: str
+ name: str
+ def __getattr__(self, name: str) -> Any: ...
+
+class UsersTable:
+ async def get_user_by_id(self, id: str, db: Any = None) -> UserModel | None: ...
+
+Users: UsersTable
diff --git a/stubs/open_webui/routers/__init__.pyi b/stubs/open_webui/routers/__init__.pyi
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/stubs/open_webui/routers/__init__.pyi
diff --git a/stubs/open_webui/routers/images.pyi b/stubs/open_webui/routers/images.pyi
new file mode 100644
index 0000000..04c759f
--- /dev/null
+++ b/stubs/open_webui/routers/images.pyi
@@ -0,0 +1,14 @@
+from types import SimpleNamespace
+from typing import Any
+
+from fastapi import Request
+
+async def get_image_config() -> SimpleNamespace: ...
+async def upload_image(
+ request: Request,
+ image_data: bytes,
+ content_type: str,
+ metadata: dict[str, Any],
+ user: Any,
+ db: Any = None,
+) -> tuple[Any, dict[str, Any]]: ...