fix: replace hardcoded tokendb path
This commit is contained in:
parent
29ee360082
commit
0b64a84e96
3 changed files with 13 additions and 4 deletions
|
|
@ -10,6 +10,4 @@ api_keys:
|
|||
"http://192.168.0.51:12434": "ollama"
|
||||
"http://192.168.0.51:12434/v1": "llama"
|
||||
|
||||
db_path: "/tmp/nomyo_test_tokens.db"
|
||||
|
||||
cache_enabled: false
|
||||
|
|
|
|||
|
|
@ -16,6 +16,7 @@ import asyncio
|
|||
import os
|
||||
import ssl
|
||||
import sys
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
|
|
@ -24,8 +25,14 @@ import httpx
|
|||
import pytest
|
||||
|
||||
_TEST_DIR = Path(__file__).parent
|
||||
# Must be set before importing router so the module-level Config.from_yaml uses test config
|
||||
# Must be set before importing router so module-level Config.from_yaml + Config field
|
||||
# defaults pick these up. db_path is intentionally absent from config_test.yaml so the
|
||||
# env-var default wins — keeps tests portable across CI runners (Linux/macOS/Windows).
|
||||
os.environ.setdefault("NOMYO_ROUTER_CONFIG_PATH", str(_TEST_DIR / "config_test.yaml"))
|
||||
os.environ.setdefault(
|
||||
"NOMYO_ROUTER_DB_PATH",
|
||||
str(Path(tempfile.gettempdir()) / "nomyo_router_test_tokens.db"),
|
||||
)
|
||||
|
||||
sys.path.insert(0, str(_TEST_DIR.parent))
|
||||
|
||||
|
|
|
|||
|
|
@ -1,4 +1,6 @@
|
|||
"""Unit tests for cache.LLMCache in exact-match mode (no sentence-transformers needed)."""
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
from types import SimpleNamespace
|
||||
|
||||
import orjson
|
||||
|
|
@ -13,6 +15,8 @@ from cache import (
|
|||
openai_nonstream_to_sse,
|
||||
)
|
||||
|
||||
_CACHE_DB_PATH = str(Path(tempfile.gettempdir()) / "nomyo_test_cache.db")
|
||||
|
||||
|
||||
def _exact_cfg(backend: str = "memory") -> SimpleNamespace:
|
||||
"""Config for exact-match mode — similarity=1.0 avoids embedding deps."""
|
||||
|
|
@ -22,7 +26,7 @@ def _exact_cfg(backend: str = "memory") -> SimpleNamespace:
|
|||
cache_similarity=1.0,
|
||||
cache_history_weight=0.3,
|
||||
cache_ttl=300,
|
||||
cache_db_path="/tmp/test_cache.db",
|
||||
cache_db_path=_CACHE_DB_PATH,
|
||||
cache_redis_url="redis://localhost:6379",
|
||||
)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue