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

fix typing

Browse Source
This commit is contained in:
Shroominic
2026-01-12 17:47:14 +08:00
parent 6199f6467b
commit 3dfbd3815c
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: