All files / src/components/settings/providers FireworksProvider.tsx

77.14% Statements 27/35
100% Branches 3/3
33.33% Functions 1/3
77.14% Lines 27/35

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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 561x   1x 1x 1x 1x 1x 1x                     1x   1x 1x 5x 5x   5x   5x 5x 5x 5x 5x 5x 5x   5x 5x 5x 5x                   5x     5x     5x  
import { fireworksModels } from "@shared/api"
import { Mode } from "@shared/storage/types"
import { useExtensionState } from "@/context/ExtensionStateContext"
import { ApiKeyField } from "../common/ApiKeyField"
import { ModelInfoView } from "../common/ModelInfoView"
import { ModelSelector } from "../common/ModelSelector"
import { normalizeApiConfiguration } from "../utils/providerUtils"
import { useApiConfigurationHandlers } from "../utils/useApiConfigurationHandlers"
 
/**
 * Props for the FireworksProvider component
 */
interface FireworksProviderProps {
	currentMode: Mode
	isPopup?: boolean
	showModelOptions: boolean
}
 
/**
 * The Fireworks provider configuration component
 */
export const FireworksProvider = ({ currentMode, isPopup, showModelOptions }: FireworksProviderProps) => {
	const { apiConfiguration } = useExtensionState()
	const { handleModeFieldChange, handleFieldChange } = useApiConfigurationHandlers()
 
	const { selectedModelId, selectedModelInfo } = normalizeApiConfiguration(apiConfiguration, currentMode)
 
	return (
		<div>
			<ApiKeyField
				initialValue={apiConfiguration?.fireworksApiKey || ""}
				onChange={(value) => handleFieldChange("fireworksApiKey", value)}
				providerName="Fireworks"
				signupUrl="https://fireworks.ai/"
			/>
			<ModelSelector
				label="Model"
				models={fireworksModels}
				onChange={(e: any) => {
					handleModeFieldChange(
						{
							plan: "planModeFireworksModelId",
							act: "actModeFireworksModelId",
						},
						e.target.value,
						currentMode,
					)
				}}
				selectedModelId={selectedModelId}
			/>
 
			<ModelInfoView isPopup={isPopup} modelInfo={selectedModelInfo} selectedModelId={selectedModelId} />
		</div>
	)
}