43 lines
1.2 KiB
TypeScript
43 lines
1.2 KiB
TypeScript
import { type Completion } from "@anthropic-ai/sdk/resources";
|
|
import { type SupportedModel } from ".";
|
|
import { type FrontendModelProvider } from "../types";
|
|
import { refinementActions } from "./refinementActions";
|
|
|
|
const frontendModelProvider: FrontendModelProvider<SupportedModel, Completion> = {
|
|
name: "Replicate Llama2",
|
|
|
|
models: {
|
|
"claude-2.0": {
|
|
name: "Claude 2.0",
|
|
contextWindow: 100000,
|
|
promptTokenPrice: 11.02 / 1000000,
|
|
completionTokenPrice: 32.68 / 1000000,
|
|
speed: "medium",
|
|
provider: "anthropic/completion",
|
|
learnMoreUrl: "https://www.anthropic.com/product",
|
|
apiDocsUrl: "https://docs.anthropic.com/claude/reference/complete_post",
|
|
},
|
|
"claude-instant-1.1": {
|
|
name: "Claude Instant 1.1",
|
|
contextWindow: 100000,
|
|
promptTokenPrice: 1.63 / 1000000,
|
|
completionTokenPrice: 5.51 / 1000000,
|
|
speed: "fast",
|
|
provider: "anthropic/completion",
|
|
learnMoreUrl: "https://www.anthropic.com/product",
|
|
apiDocsUrl: "https://docs.anthropic.com/claude/reference/complete_post",
|
|
},
|
|
},
|
|
|
|
refinementActions,
|
|
|
|
normalizeOutput: (output) => {
|
|
return {
|
|
type: "text",
|
|
value: output.completion,
|
|
};
|
|
},
|
|
};
|
|
|
|
export default frontendModelProvider;
|