File size: 4,857 Bytes
98b7f49 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 |
// API client for AnyCoder backend
import axios, { AxiosInstance } from 'axios';
import type {
Model,
AuthStatus,
CodeGenerationRequest,
DeploymentRequest,
DeploymentResponse,
Language,
} from '@/types';
// Use relative URLs in production (Next.js rewrites will proxy to backend)
// In local dev without rewrites, use localhost:8000
const API_URL = process.env.NEXT_PUBLIC_API_URL ||
(typeof window !== 'undefined' && window.location.hostname !== 'localhost'
? '' // Use relative URLs in production (proxied by Next.js)
: 'http://localhost:8000'); // Local development
class ApiClient {
private client: AxiosInstance;
private token: string | null = null;
constructor() {
this.client = axios.create({
baseURL: API_URL,
headers: {
'Content-Type': 'application/json',
},
});
// Add auth token to requests if available
this.client.interceptors.request.use((config) => {
if (this.token) {
config.headers.Authorization = `Bearer ${this.token}`;
}
return config;
});
// Load token from localStorage on client side
if (typeof window !== 'undefined') {
this.token = localStorage.getItem('hf_oauth_token');
}
}
setToken(token: string | null) {
this.token = token;
// Note: OAuth token is stored by auth.ts, not here
// We just keep it in memory for API calls
}
getToken(): string | null {
return this.token;
}
async getModels(): Promise<Model[]> {
const response = await this.client.get<Model[]>('/api/models');
return response.data;
}
async getLanguages(): Promise<{ languages: Language[] }> {
const response = await this.client.get<{ languages: Language[] }>('/api/languages');
return response.data;
}
async getAuthStatus(): Promise<AuthStatus> {
try {
const response = await this.client.get<AuthStatus>('/api/auth/status');
return response.data;
} catch (error) {
return {
authenticated: false,
message: 'Not authenticated',
};
}
}
// Stream-based code generation using EventSource (Server-Sent Events)
generateCodeStream(
request: CodeGenerationRequest,
onChunk: (content: string) => void,
onComplete: (code: string) => void,
onError: (error: string) => void
): () => void {
const url = new URL('/api/generate', API_URL);
const eventSource = new EventSource(
url.toString() + '?' + new URLSearchParams({
query: request.query,
language: request.language,
model_id: request.model_id,
provider: request.provider,
})
);
eventSource.onmessage = (event) => {
try {
const data = JSON.parse(event.data);
if (data.type === 'chunk' && data.content) {
onChunk(data.content);
} else if (data.type === 'complete' && data.code) {
onComplete(data.code);
eventSource.close();
} else if (data.type === 'error') {
onError(data.message || 'Unknown error occurred');
eventSource.close();
}
} catch (error) {
console.error('Error parsing SSE data:', error);
}
};
eventSource.onerror = (error) => {
console.error('EventSource error:', error);
onError('Connection error occurred');
eventSource.close();
};
// Return cleanup function
return () => {
eventSource.close();
};
}
// Alternative: WebSocket-based generation
generateCodeWebSocket(
request: CodeGenerationRequest,
onChunk: (content: string) => void,
onComplete: (code: string) => void,
onError: (error: string) => void
): WebSocket {
const wsUrl = API_URL.replace('http', 'ws') + '/ws/generate';
const ws = new WebSocket(wsUrl);
ws.onopen = () => {
ws.send(JSON.stringify(request));
};
ws.onmessage = (event) => {
try {
const data = JSON.parse(event.data);
if (data.type === 'chunk' && data.content) {
onChunk(data.content);
} else if (data.type === 'complete' && data.code) {
onComplete(data.code);
ws.close();
} else if (data.type === 'error') {
onError(data.message || 'Unknown error occurred');
ws.close();
}
} catch (error) {
console.error('Error parsing WebSocket data:', error);
}
};
ws.onerror = (error) => {
console.error('WebSocket error:', error);
onError('Connection error occurred');
};
return ws;
}
async deploy(request: DeploymentRequest): Promise<DeploymentResponse> {
const response = await this.client.post<DeploymentResponse>('/api/deploy', request);
return response.data;
}
logout() {
this.token = null;
}
}
// Export singleton instance
export const apiClient = new ApiClient();
|