laantungir/routstr-core
1
0
Fork 0
mirror of https://github.com/Routstr/routstr-core.git synced 2026-08-10 11:13:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typing

Browse Source
This commit is contained in:
Shroominic
2025-12-20 14:02:56 +01:00
parent ec0fcfb48b
commit d41c214d9e
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
routstr/upstream/fake.py
+6 -2
View File
@@ -43,7 +43,11 @@ class MockUpstreamProvider(BaseUpstreamProvider):
created = 1766138895
model = "mock/gpt-420-mock"
def make_chunk(delta: dict, finish_reason=None, usage=None):
def make_chunk(
delta: dict,
finish_reason: str | None = None,
usage: dict | None = None,
) -> bytes:
chunk = {
"id": req_id,
"provider": "MockProvider",
@@ -254,7 +258,7 @@ class MockUpstreamProvider(BaseUpstreamProvider):
),
]
async def transform_model_name(self, model_id: str) -> str:
def transform_model_name(self, model_id: str) -> str:
return "fake-model"
async def get_balance(self) -> float | None: