milvusConfig.ts•3.26 kB
export interface MilvusConfig {
address: string;
token?: string;
username?: string;
password?: string;
database?: string;
}
export interface ChromeStorageConfig {
githubToken?: string;
openaiToken?: string;
milvusAddress?: string;
milvusToken?: string;
milvusUsername?: string;
milvusPassword?: string;
milvusDatabase?: string;
}
export class MilvusConfigManager {
/**
* Get Milvus configuration from Chrome storage
*/
static async getMilvusConfig(): Promise<MilvusConfig | null> {
return new Promise((resolve) => {
chrome.storage.sync.get([
'milvusAddress',
'milvusToken',
'milvusUsername',
'milvusPassword',
'milvusDatabase'
], (items: ChromeStorageConfig) => {
if (chrome.runtime.lastError) {
console.error('Error loading Milvus config:', chrome.runtime.lastError);
resolve(null);
return;
}
if (!items.milvusAddress) {
resolve(null);
return;
}
const config: MilvusConfig = {
address: items.milvusAddress,
token: items.milvusToken,
username: items.milvusUsername,
password: items.milvusPassword,
database: items.milvusDatabase || 'default'
};
resolve(config);
});
});
}
/**
* Save Milvus configuration to Chrome storage
*/
static async saveMilvusConfig(config: MilvusConfig): Promise<void> {
return new Promise((resolve, reject) => {
chrome.storage.sync.set({
milvusAddress: config.address,
milvusToken: config.token,
milvusUsername: config.username,
milvusPassword: config.password,
milvusDatabase: config.database || 'default'
}, () => {
if (chrome.runtime.lastError) {
reject(chrome.runtime.lastError);
} else {
resolve();
}
});
});
}
/**
* Get OpenAI configuration
*/
static async getOpenAIConfig(): Promise<{ apiKey: string; model: string } | null> {
return new Promise((resolve) => {
chrome.storage.sync.get(['openaiToken'], (items: ChromeStorageConfig) => {
if (chrome.runtime.lastError || !items.openaiToken) {
resolve(null);
return;
}
resolve({
apiKey: items.openaiToken,
model: 'text-embedding-3-small' // Default model
});
});
});
}
/**
* Validate Milvus configuration
*/
static validateMilvusConfig(config: MilvusConfig): boolean {
if (!config.address) {
return false;
}
// For basic validation, just check if address is provided
// Authentication can be optional for local instances
return true;
}
}