onboarding before tests

This commit is contained in:
phact 2025-12-19 03:59:37 -05:00
parent 304e0c15d2
commit 3c14905440
2 changed files with 68 additions and 0 deletions

View file

@ -22,6 +22,42 @@ pytestmark = pytest.mark.skipif(
# Module-level cache for API key (created once, reused)
_cached_api_key: str | None = None
_base_url = os.environ.get("OPENRAG_URL", "http://localhost:3000")
_onboarding_done = False
@pytest.fixture(scope="session", autouse=True)
def ensure_onboarding():
"""Ensure the OpenRAG instance is onboarded before running tests.
This marks the config as 'edited' so that settings updates are allowed.
"""
global _onboarding_done
if _onboarding_done:
return
onboarding_payload = {
"llm_provider": "openai",
"embedding_provider": "openai",
"embedding_model": "text-embedding-3-small",
"llm_model": "gpt-4o-mini",
"sample_data": False,
}
try:
response = httpx.post(
f"{_base_url}/api/onboarding",
json=onboarding_payload,
timeout=30.0,
)
if response.status_code in (200, 204):
print(f"[SDK Tests] Onboarding completed successfully")
else:
# May already be onboarded, which is fine
print(f"[SDK Tests] Onboarding returned {response.status_code}: {response.text[:200]}")
except Exception as e:
print(f"[SDK Tests] Onboarding request failed: {e}")
_onboarding_done = True
def get_api_key() -> str:

View file

@ -18,6 +18,35 @@ let OpenRAGClient: typeof import("../src").OpenRAGClient;
const BASE_URL = process.env.OPENRAG_URL || "http://localhost:3000";
const SKIP_TESTS = process.env.SKIP_SDK_INTEGRATION_TESTS === "true";
// Ensure the OpenRAG instance is onboarded before running tests
async function ensureOnboarding(): Promise<void> {
const onboardingPayload = {
llm_provider: "openai",
embedding_provider: "openai",
embedding_model: "text-embedding-3-small",
llm_model: "gpt-4o-mini",
sample_data: false,
};
try {
const response = await fetch(`${BASE_URL}/api/onboarding`, {
method: "POST",
headers: { "Content-Type": "application/json" },
body: JSON.stringify(onboardingPayload),
});
if (response.status === 200 || response.status === 204) {
console.log("[SDK Tests] Onboarding completed successfully");
} else {
// May already be onboarded, which is fine
const text = await response.text();
console.log(`[SDK Tests] Onboarding returned ${response.status}: ${text.slice(0, 200)}`);
}
} catch (e) {
console.log(`[SDK Tests] Onboarding request failed: ${e}`);
}
}
// Create API key for tests
async function createApiKey(): Promise<string> {
// Use /api/keys to go through frontend proxy (frontend at :3000 proxies /api/* to backend)
@ -57,6 +86,9 @@ describe.skipIf(SKIP_TESTS)("OpenRAG TypeScript SDK Integration", () => {
let testFilePath: string;
beforeAll(async () => {
// Ensure onboarding is done first (marks config as edited)
await ensureOnboarding();
// Import SDK
const sdk = await import("../src");
OpenRAGClient = sdk.OpenRAGClient;