replicate/llama2 provider

Still need to fix the types but it runs
This commit is contained in:
Kyle Corbitt
2023-07-20 19:55:03 -07:00
parent 332a2101c0
commit 847753c32b
14 changed files with 202 additions and 29 deletions

View File

@@ -70,6 +70,7 @@ export default async function parseConstructFn(
// We've validated the JSON schema so this should be safe
const input = prompt.input as Parameters<(typeof provider)["getModel"]>[0];
// @ts-expect-error TODO FIX ASAP
const model = provider.getModel(input);
if (!model) {
return {
@@ -79,6 +80,8 @@ export default async function parseConstructFn(
return {
modelProvider: prompt.modelProvider as keyof typeof modelProviders,
// @ts-expect-error TODO FIX ASAP
model,
modelInput: input,
};