Base code

frontend/src/api/onboarding.ts

@@ -0,0 +1,214 @@
+// Make sure to install axios: npm install axios
+import { AxiosResponse } from 'axios';
+import { apiClient } from './client';
+
+export interface APIKeyRequest {
+  provider: string;
+  api_key: string;
+  description?: string;
+}
+
+export interface APIKeyResponse {
+  provider: string;
+  api_key: string;
+  description?: string;
+}
+
+export interface OnboardingStepResponse {
+  step: number;
+  data?: any;
+  validation_errors?: string[];
+}
+
+export interface OnboardingSessionResponse {
+  id: number;
+  user_id: number;
+  current_step: number;
+  progress: number;
+}
+
+export interface OnboardingProgressResponse {
+  progress: number;
+  current_step: number;
+  total_steps: number;
+  completion_percentage: number;
+}
+
+export async function startOnboarding() {
+  const res: AxiosResponse<OnboardingSessionResponse> = await apiClient.post('/api/onboarding/start');
+  return res.data;
+}
+
+export async function getCurrentStep() {
+  // Get the current step from the onboarding status
+  console.log('getCurrentStep: Calling /api/onboarding/status');
+  const res: AxiosResponse<any> = await apiClient.get('/api/onboarding/status');
+  console.log('getCurrentStep: Backend returned:', res.data);
+  return { step: res.data.current_step || 1 };
+}
+
+export async function setCurrentStep(step: number, stepData?: any) {
+  // Complete the current step to move to the next one
+  console.log('setCurrentStep: Completing step', step, 'with data:', stepData);
+  const res: AxiosResponse<OnboardingStepResponse> = await apiClient.post(`/api/onboarding/step/${step}/complete`, {
+    data: stepData || {},
+    validation_errors: []
+  });
+  console.log('setCurrentStep: Backend response:', res.data);
+  return { step };
+}
+
+export async function getApiKeys() {
+  const maxRetries = 3;
+  let lastError: any;
+  
+  console.log('getApiKeys: Starting API call to /api/onboarding/api-keys');
+  
+  for (let attempt = 0; attempt < maxRetries; attempt++) {
+    try {
+      console.log(`getApiKeys: Attempt ${attempt + 1}/${maxRetries}`);
+      const res: AxiosResponse<Record<string, string>> = await apiClient.get('/api/onboarding/api-keys');
+      console.log('getApiKeys: API call successful');
+      return res.data;
+    } catch (error: any) {
+      lastError = error;
+      console.log(`getApiKeys: Attempt ${attempt + 1} failed:`, error.response?.status, error.message);
+      
+      // If it's a rate limit error (429), wait and retry
+      if (error.response?.status === 429) {
+        const retryAfter = error.response?.data?.retry_after || 60;
+        const delay = Math.min(retryAfter * 1000, 5000); // Max 5 seconds
+        
+        console.log(`getApiKeys: Rate limited, retrying in ${delay}ms (attempt ${attempt + 1}/${maxRetries})`);
+        await new Promise(resolve => setTimeout(resolve, delay));
+        continue;
+      }
+      
+      // For other errors, don't retry
+      console.log('getApiKeys: Non-rate-limit error, not retrying');
+      throw error;
+    }
+  }
+  
+  // If we've exhausted all retries, throw the last error
+  console.log('getApiKeys: All retries exhausted');
+  throw lastError;
+}
+
+export async function getApiKeysForOnboarding() {
+  const maxRetries = 3;
+  let lastError: any;
+  
+  console.log('getApiKeysForOnboarding: Starting API call to /api/onboarding/api-keys/onboarding');
+  
+  for (let attempt = 0; attempt < maxRetries; attempt++) {
+    try {
+      console.log(`getApiKeysForOnboarding: Attempt ${attempt + 1}/${maxRetries}`);
+      const res: AxiosResponse<any> = await apiClient.get('/api/onboarding/api-keys/onboarding');
+      console.log('getApiKeysForOnboarding: API call successful');
+      return res.data.api_keys || {};
+    } catch (error: any) {
+      lastError = error;
+      console.log(`getApiKeysForOnboarding: Attempt ${attempt + 1} failed:`, error.response?.status, error.message);
+      
+      // If it's a rate limit error (429), wait and retry
+      if (error.response?.status === 429) {
+        const retryAfter = error.response?.data?.retry_after || 60;
+        const delay = Math.min(retryAfter * 1000, 5000); // Max 5 seconds
+        
+        console.log(`getApiKeysForOnboarding: Rate limited, retrying in ${delay}ms (attempt ${attempt + 1}/${maxRetries})`);
+        await new Promise(resolve => setTimeout(resolve, delay));
+        continue;
+      }
+      
+      // For other errors, don't retry
+      console.log('getApiKeysForOnboarding: Non-rate-limit error, not retrying');
+      throw error;
+    }
+  }
+  
+  // If we've exhausted all retries, throw the last error
+  console.log('getApiKeysForOnboarding: All retries exhausted');
+  throw lastError;
+}
+
+export async function saveApiKey(provider: string, api_key: string, description?: string) {
+  const res: AxiosResponse<APIKeyResponse> = await apiClient.post('/api/onboarding/api-keys', { 
+    provider, 
+    api_key,
+    description 
+  });
+  return res.data;
+}
+
+export async function getProgress() {
+  const res: AxiosResponse<OnboardingProgressResponse> = await apiClient.get('/api/onboarding/progress');
+  return { progress: res.data.completion_percentage || 0 };
+}
+
+export async function setProgress(progress: number) {
+  // Progress is managed automatically by the backend
+  // This function is kept for compatibility but doesn't make a backend call
+  return { progress };
+}
+
+// Additional functions for better integration
+export async function getOnboardingConfig() {
+  const res: AxiosResponse<any> = await apiClient.get('/api/onboarding/config');
+  return res.data;
+}
+
+export async function getStepData(stepNumber: number) {
+  const res: AxiosResponse<any> = await apiClient.get(`/api/onboarding/step/${stepNumber}`);
+  return res.data;
+}
+
+export async function getStep1ApiKeysFromProgress(): Promise<{ gemini?: string; exa?: string; copilotkit?: string }> {
+  try {
+    const step = await getStepData(1);
+    const keys = step?.data?.api_keys || {};
+    return {
+      gemini: keys.gemini || undefined,
+      exa: keys.exa || undefined,
+      copilotkit: keys.copilotkit || undefined,
+    };
+  } catch (_e) {
+    return {};
+  }
+}
+
+export async function skipStep(stepNumber: number) {
+  const res: AxiosResponse<any> = await apiClient.post(`/api/onboarding/step/${stepNumber}/skip`);
+  return res.data;
+}
+
+export async function validateApiKeys() {
+  const res: AxiosResponse<any> = await apiClient.post('/api/onboarding/api-keys/validate');
+  return res.data;
+}
+
+export async function completeOnboarding() {
+  const res: AxiosResponse<any> = await apiClient.post('/api/onboarding/complete');
+  return res.data;
+}
+
+export async function resetOnboarding() {
+  const res: AxiosResponse<any> = await apiClient.post('/api/onboarding/reset');
+  return res.data;
+}
+
+// New functions for FinalStep data loading
+export async function getOnboardingSummary() {
+  const res: AxiosResponse<any> = await apiClient.get('/api/onboarding/summary');
+  return res.data;
+}
+
+export async function getWebsiteAnalysisData() {
+  const res: AxiosResponse<any> = await apiClient.get('/api/onboarding/website-analysis');
+  return res.data;
+}
+
+export async function getResearchPreferencesData() {
+  const res: AxiosResponse<any> = await apiClient.get('/api/onboarding/research-preferences');
+  return res.data;
+}