@mariozechner/pi-web-ui 0.5.44 → 0.5.45

package/src/components/ProviderKeyInput.ts

@@ -35,7 +35,7 @@ export class ProviderKeyInput extends LitElement {
 
 	private async checkKeyStatus() {
 		try {
-			const key = await getAppStorage().providerKeys.getKey(this.provider);
+			const key = await getAppStorage().providerKeys.get(this.provider);
 			this.hasKey = !!key;
 		} catch (error) {
 			console.error("Failed to check key status:", error);
@@ -45,7 +45,8 @@ export class ProviderKeyInput extends LitElement {
 	private async testApiKey(provider: string, apiKey: string): Promise<boolean> {
 		try {
 			const modelId = TEST_MODELS[provider];
-			if (!modelId) return false;
+			// Returning true here for Ollama and friends. Can' know which model to use for testing
+			if (!modelId) return true;
 
 			let model = getModel(provider as any, modelId);
 			if (!model) return false;
@@ -67,7 +68,7 @@ export class ProviderKeyInput extends LitElement {
 
 			const result = await complete(model, context, {
 				apiKey,
-				maxTokens: 10,
+				maxTokens: 200,
 			} as any);
 
 			return result.stopReason === "stop";
@@ -89,7 +90,7 @@ export class ProviderKeyInput extends LitElement {
 
 		if (success) {
 			try {
-				await getAppStorage().providerKeys.setKey(this.provider, this.keyInput);
+				await getAppStorage().providerKeys.set(this.provider, this.keyInput);
 				this.hasKey = true;
 				this.keyInput = "";
 				this.requestUpdate();
@@ -112,7 +113,7 @@ export class ProviderKeyInput extends LitElement {
 
 	private async removeKey() {
 		try {
-			await getAppStorage().providerKeys.removeKey(this.provider);
+			await getAppStorage().providerKeys.delete(this.provider);
 			this.hasKey = false;
 			this.keyInput = "";
 			this.requestUpdate();