Skip to content

Commit 1fba27b

Browse files
committed
Fixes API response for OpenWebUI
1 parent c24e6d5 commit 1fba27b

File tree

1 file changed

+17
-19
lines changed

1 file changed

+17
-19
lines changed

src/dialog/routers/openai.py

+17-19
Original file line numberDiff line numberDiff line change
@@ -26,25 +26,23 @@ async def get_models():
2626
Returns the model that is available inside Dialog in the OpenAI format.
2727
"""
2828

29-
return {
30-
"models": [
31-
OpenAIModel(**{
32-
"id": "talkd-ai",
33-
"object": "model",
34-
"created": int(datetime.datetime.now().timestamp()),
35-
"owned_by": "system",
36-
"digest": str(uuid4())
37-
})
38-
] + [
39-
OpenAIModel(**{
40-
"id": model["model_name"],
41-
"object": "model",
42-
"created": int(datetime.datetime.now().timestamp()),
43-
"owned_by": "system",
44-
"digest": str(uuid4())
45-
}) for model in Settings().PROJECT_CONFIG.get("endpoint", [])
46-
]
47-
}
29+
return [
30+
OpenAIModel(**{
31+
"id": "talkd-ai",
32+
"object": "model",
33+
"created": int(datetime.datetime.now().timestamp()),
34+
"owned_by": "system",
35+
"digest": str(uuid4())
36+
})
37+
] + [
38+
OpenAIModel(**{
39+
"id": model["model_name"],
40+
"object": "model",
41+
"created": int(datetime.datetime.now().timestamp()),
42+
"owned_by": "system",
43+
"digest": str(uuid4())
44+
}) for model in Settings().PROJECT_CONFIG.get("endpoint", [])
45+
]
4846

4947
@open_ai_api_router.get("/api/tags")
5048
async def get_tags():

0 commit comments

Comments
 (0)