remove the watsonx models which does not support text_chat

Author: PrasangAPrajapati

src/api/providers/__tests__/watsonx.spec.ts

@@ -192,7 +192,7 @@ describe("WatsonxAIHandler", () => {
 				chunks.push(chunk)
 			}
 
-			expect(chunks.length).toBe(1)
+			expect(chunks.length).toBe(2)
 			expect(chunks[0]).toEqual({
 				type: "text",
 				text: testContent,

src/api/providers/fetchers/watsonx.ts

@@ -60,15 +60,21 @@ export async function getWatsonxModels(
 		let knownModels: Record<string, ModelInfo> = {}
 
 		try {
-			const response = await service.listFoundationModelSpecs({ filters: "!function_embedding" })
+			const response = await service.listFoundationModelSpecs({ filters: "function_text_chat" })
 			if (response && response.result) {
 				const result = response.result as any
 				const modelsList = result.resources
 				if (Array.isArray(modelsList) && modelsList.length > 0) {
 					for (const model of modelsList) {
 						const modelId = model.id || model.name || model.model_id
-						const contextWindow = model.model_limits.max_sequence_length || 131072
-						const maxTokens = model.model_limits.max_output_tokens || Math.floor(contextWindow / 2)
+						let contextWindow = 131072
+						if (model.model_limits && model.model_limits.max_sequence_length) {
+							contextWindow = model.model_limits.max_sequence_length
+						}
+						let maxTokens = Math.floor(contextWindow / 2)
+						if (model.model_limits && model.model_limits.max_output_tokens) {
+							maxTokens = model.model_limits.max_output_tokens
+						}
 
 						let description = ""
 						if (model.long_description) {

src/api/providers/watsonx.ts

@@ -9,6 +9,7 @@ import type { SingleCompletionHandler, ApiHandlerCreateMessageMetadata } from ".
 import { WatsonXAI } from "@ibm-cloud/watsonx-ai"
 import { convertToWatsonxAiMessages } from "../transform/watsonxai-format"
 import OpenAI from "openai"
+import { calculateApiCostOpenAI } from "../../shared/cost"
 
 export class WatsonxAIHandler extends BaseProvider implements SingleCompletionHandler {
 	private options: ApiHandlerOptions
@@ -154,6 +155,20 @@ export class WatsonxAIHandler extends BaseProvider implements SingleCompletionHa
 				type: "text",
 				text: responseText,
 			}
+			let usageInfo: any = null
+			usageInfo = response.result.usage || {}
+			const outputTokens = usageInfo.completion_tokens
+
+			const inputTokens = usageInfo?.prompt_tokens || 0
+			const modelInfo = this.getModel().info
+			const totalCost = calculateApiCostOpenAI(modelInfo, inputTokens, outputTokens)
+
+			yield {
+				type: "usage",
+				inputTokens: inputTokens,
+				outputTokens,
+				totalCost: totalCost,
+			}
 		} catch (error) {
 			// Extract error message and type from the error object
 			const errorMessage = error?.message || String(error)

webview-ui/src/components/settings/ApiOptions.tsx

@@ -778,6 +778,7 @@ const ApiOptions = ({
 						<DiffSettingsControl
 							diffEnabled={apiConfiguration.diffEnabled}
 							fuzzyMatchThreshold={apiConfiguration.fuzzyMatchThreshold}
+							provider={apiConfiguration.apiProvider}
 							onChange={(field, value) => setApiConfigurationField(field, value)}
 						/>
 						{selectedModelInfo?.supportsTemperature !== false && (

webview-ui/src/components/settings/DiffSettingsControl.tsx

@@ -6,12 +6,14 @@ import { VSCodeCheckbox } from "@vscode/webview-ui-toolkit/react"
 interface DiffSettingsControlProps {
 	diffEnabled?: boolean
 	fuzzyMatchThreshold?: number
+	provider: string | undefined
 	onChange: (field: "diffEnabled" | "fuzzyMatchThreshold", value: any) => void
 }
 
 export const DiffSettingsControl: React.FC<DiffSettingsControlProps> = ({
 	diffEnabled = true,
 	fuzzyMatchThreshold = 1.0,
+	provider,
 	onChange,
 }) => {
 	const { t } = useAppTranslation()
@@ -37,7 +39,9 @@ export const DiffSettingsControl: React.FC<DiffSettingsControlProps> = ({
 					<span className="font-medium">{t("settings:advanced.diff.label")}</span>
 				</VSCodeCheckbox>
 				<div className="text-vscode-descriptionForeground text-sm">
-					{t("settings:advanced.diff.description")}
+					{provider === "watsonx"
+						? t("settings:advanced.diff.watsonx.description")
+						: t("settings:advanced.diff.description")}
 				</div>
 			</div>
 

webview-ui/src/i18n/locales/ca/settings.json

@@ -688,6 +688,9 @@
 					"standard": "Standard diff strategy applies changes to a single code block at a time.",
 					"unified": "Unified diff strategy takes multiple approaches to applying diffs and chooses the best approach.",
 					"multiBlock": "Multi-block diff strategy allows updating multiple code blocks in a file in one request."
+				},
+				"watsonx": {
+					"description": "When enabled, Roo will be able to edit files more quickly and will automatically reject truncated full-file writes."
 				}
 			},
 			"matchPrecision": {