Replace EverMemOS with EverOS backend
This commit is contained in:
@ -7,7 +7,7 @@ from fastapi import FastAPI
|
||||
from httpx import ASGITransport, AsyncClient
|
||||
|
||||
import memory_gateway.api_v2 as api_v2
|
||||
from memory_gateway.evermemos_client import EverMemOSClient
|
||||
from memory_gateway.everos_client import EverOSClient
|
||||
from memory_gateway.openviking_client import OpenVikingClient
|
||||
from memory_gateway.repositories import InMemoryRepository
|
||||
from memory_gateway.schemas_v2 import BackendRefStatus, BackendType, IngestRequest, IngestResponse, OperationStatus
|
||||
@ -28,13 +28,13 @@ def _env(name: str) -> str:
|
||||
return value
|
||||
|
||||
|
||||
def test_real_openviking_and_evermemos_ingest_writes_memory_refs():
|
||||
def test_real_openviking_and_everos_ingest_writes_memory_refs():
|
||||
openviking_base_url = _env("OPENVIKING_BASE_URL")
|
||||
evermemos_base_url = _env("EVERMEMOS_BASE_URL")
|
||||
everos_base_url = _env("EVEROS_BASE_URL")
|
||||
openviking_api_key = os.environ.get("OPENVIKING_API_KEY", "")
|
||||
evermemos_api_key = os.environ.get("EVERMEMOS_API_KEY", "")
|
||||
everos_api_key = os.environ.get("EVEROS_API_KEY", "")
|
||||
openviking_ingest_path = os.environ.get("OPENVIKING_INGEST_PATH")
|
||||
evermemos_ingest_path = os.environ.get("EVERMEMOS_INGEST_PATH")
|
||||
everos_ingest_path = os.environ.get("EVEROS_INGEST_PATH")
|
||||
|
||||
async def openviking_factory():
|
||||
return OpenVikingClient(
|
||||
@ -48,11 +48,11 @@ def test_real_openviking_and_evermemos_ingest_writes_memory_refs():
|
||||
service = MemoryGatewayV2Service(
|
||||
repo=repo,
|
||||
openviking_client_factory=openviking_factory,
|
||||
evermemos_client=EverMemOSClient(
|
||||
everos_client=EverOSClient(
|
||||
mode="real",
|
||||
base_url=evermemos_base_url,
|
||||
api_key=evermemos_api_key,
|
||||
ingest_path=evermemos_ingest_path,
|
||||
base_url=everos_base_url,
|
||||
api_key=everos_api_key,
|
||||
ingest_path=everos_ingest_path,
|
||||
),
|
||||
)
|
||||
run_id = uuid4().hex[:12]
|
||||
@ -60,20 +60,20 @@ def test_real_openviking_and_evermemos_ingest_writes_memory_refs():
|
||||
response = asyncio.run(post_ingest(service, run_id))
|
||||
|
||||
refs = repo.list_memory_refs(session_id=f"real_ingest_sess_{run_id}", limit=10)
|
||||
assert {ref.backend_type for ref in refs} == {BackendType.OPENVIKING, BackendType.EVERMEMOS}
|
||||
assert {ref.backend_type for ref in refs} == {BackendType.OPENVIKING, BackendType.EVEROS}
|
||||
assert all(ref.content_hash for ref in refs)
|
||||
openviking_ref = next(ref for ref in refs if ref.backend_type == BackendType.OPENVIKING)
|
||||
evermemos_ref = next(ref for ref in refs if ref.backend_type == BackendType.EVERMEMOS)
|
||||
everos_ref = next(ref for ref in refs if ref.backend_type == BackendType.EVEROS)
|
||||
|
||||
assert openviking_ref.status == BackendRefStatus.SUCCESS
|
||||
if evermemos_ref.status == BackendRefStatus.SUCCESS:
|
||||
if everos_ref.status == BackendRefStatus.SUCCESS:
|
||||
assert response.status == OperationStatus.SUCCESS
|
||||
assert evermemos_ref.native_id
|
||||
assert evermemos_ref.native_uri
|
||||
assert everos_ref.native_id
|
||||
assert everos_ref.native_uri
|
||||
else:
|
||||
assert evermemos_ref.status == BackendRefStatus.FAILED
|
||||
assert everos_ref.status == BackendRefStatus.FAILED
|
||||
assert response.status == OperationStatus.PARTIAL_SUCCESS
|
||||
assert evermemos_ref.error_message
|
||||
assert everos_ref.error_message
|
||||
|
||||
|
||||
async def post_ingest(service: MemoryGatewayV2Service, run_id: str):
|
||||
|
||||
Reference in New Issue
Block a user