Skip to content

Commit c24e6d5

Browse files
committed
Fix compatibility issue with older memory provider
1 parent 562a214 commit c24e6d5

File tree

5 files changed

+67
-52
lines changed

5 files changed

+67
-52
lines changed

docker-compose-open-webui.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
version: '3.3'
21
services:
32
db:
43
image: pgvector/pgvector:pg15

docker-compose.dev-container.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
version: "3.3"
21
services:
32
db:
43
image: pgvector/pgvector:pg15

poetry.lock

Lines changed: 37 additions & 35 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,15 +16,15 @@ uvicorn = { extras = ["standard"], version = "0.24.0.post1" }
1616
psycopg2 = "2.9.9"
1717
python-decouple = "3.8"
1818
pydantic = "2.7.4"
19-
psycopg = "3.1.12"
19+
psycopg = "3.2.2"
2020
tiktoken = "<1"
2121
pandas = "2.1.3"
2222
scikit-learn = "1.5.0"
2323
alembic = "1.12.1"
2424
importlib-metadata = "7.0.1"
2525
pyarrow = "15.0.0"
2626
iniconfig = "2.0.0"
27-
dialog-lib = "0.0.2.7"
27+
dialog-lib = "0.0.3.0"
2828
langserve = "0.2.0"
2929
sse-starlette = "2.1.0"
3030

src/dialog/routers/openai.py

Lines changed: 28 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -26,19 +26,34 @@ async def get_models():
2626
Returns the model that is available inside Dialog in the OpenAI format.
2727
"""
2828

29-
return [OpenAIModel(**{
30-
"id": "talkd-ai",
31-
"object": "model",
32-
"created": int(datetime.datetime.now().timestamp()),
33-
"owned_by": "system"
34-
})] + [
35-
OpenAIModel(**{
36-
"id": model["model_name"],
37-
"object": "model",
38-
"created": int(datetime.datetime.now().timestamp()),
39-
"owned_by": "system"
40-
}) for model in Settings().PROJECT_CONFIG.get("endpoint", [])
41-
]
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+
}
48+
49+
@open_ai_api_router.get("/api/tags")
50+
async def get_tags():
51+
"""
52+
Returns the tags that are available inside Dialog in the OpenAI format.
53+
"""
54+
55+
return await get_models()
56+
4257

4358
@open_ai_api_router.post("/chat/completions")
4459
async def ask_question_to_llm(message: OpenAIChat, session: Session = Depends(get_session)):

0 commit comments

Comments
 (0)