mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-06-29 15:31:05 +08:00
Revert "Fix: /openai/<chat_id>/chat/completions not aware of session_id" (#15205)
Reverts infiniflow/ragflow#15155 because this is never supported, keep it as it is.
This commit is contained in:
@@ -140,60 +140,6 @@ def test_openai_compatible_nonstream_shape(rest_client, create_chat):
|
||||
assert usage["total_tokens"] == usage["prompt_tokens"] + usage["completion_tokens"], usage
|
||||
|
||||
|
||||
@pytest.mark.p2
|
||||
def test_openai_compatible_nonstream_supports_session_id(rest_client, create_chat):
|
||||
chat_id = create_chat("restful_openai_session_chat")
|
||||
create_session = rest_client.post(f"/chats/{chat_id}/sessions", json={"name": "openai_session"})
|
||||
assert create_session.status_code == 200
|
||||
create_session_payload = create_session.json()
|
||||
assert create_session_payload["code"] == 0, create_session_payload
|
||||
session_id = create_session_payload["data"]["id"]
|
||||
|
||||
res = rest_client.post(
|
||||
f"/openai/{chat_id}/chat/completions",
|
||||
json={
|
||||
"model": "model",
|
||||
"messages": [{"role": "user", "content": "remember this turn"}],
|
||||
"stream": False,
|
||||
"session_id": session_id,
|
||||
},
|
||||
timeout=60,
|
||||
)
|
||||
assert res.status_code == 200
|
||||
payload = res.json()
|
||||
assert payload["session_id"] == session_id, payload
|
||||
assert payload["object"] == "chat.completion", payload
|
||||
|
||||
get_session = rest_client.get(f"/chats/{chat_id}/sessions/{session_id}")
|
||||
assert get_session.status_code == 200
|
||||
session_payload = get_session.json()
|
||||
assert session_payload["code"] == 0, session_payload
|
||||
messages = session_payload["data"]["messages"]
|
||||
assert any(
|
||||
message["role"] == "user" and message["content"] == "remember this turn"
|
||||
for message in messages
|
||||
), messages
|
||||
assert any(message["role"] == "assistant" for message in messages), messages
|
||||
|
||||
|
||||
@pytest.mark.p2
|
||||
def test_openai_compatible_invalid_session_id(rest_client, create_chat):
|
||||
chat_id = create_chat("restful_openai_invalid_session_chat")
|
||||
res = rest_client.post(
|
||||
f"/openai/{chat_id}/chat/completions",
|
||||
json={
|
||||
"model": "model",
|
||||
"messages": [{"role": "user", "content": "hello"}],
|
||||
"stream": False,
|
||||
"session_id": "invalid_session",
|
||||
},
|
||||
)
|
||||
assert res.status_code == 200
|
||||
payload = res.json()
|
||||
assert payload["code"] != 0, payload
|
||||
assert "Session not found!" in payload["message"], payload
|
||||
|
||||
|
||||
@pytest.mark.p2
|
||||
def test_openai_compatible_nonstream_with_reference_output_shape(rest_client, create_chat):
|
||||
chat_id = create_chat("restful_openai_reference_chat")
|
||||
|
||||
Reference in New Issue
Block a user