Spaces:
Paused
Paused
matt HOFFNER
commited on
Commit
·
0e460a1
1
Parent(s):
fc46386
fix
Browse files- pages/api/chat/index.ts +2 -1
pages/api/chat/index.ts
CHANGED
@@ -2,6 +2,7 @@ import { OpenAIStream, StreamingTextResponse } from "ai";
|
|
2 |
import { Configuration, OpenAIApi } from "openai-edge";
|
3 |
import { HfInference } from '@huggingface/inference';
|
4 |
import { HuggingFaceStream } from 'ai';
|
|
|
5 |
import { LLMStream } from "../../../utils/llm";
|
6 |
|
7 |
export const runtime = 'edge';
|
@@ -38,7 +39,7 @@ export default async function(req: Request) {
|
|
38 |
const response = Hf.textGenerationStream({
|
39 |
// @ts-ignore
|
40 |
model: aiProvider,
|
41 |
-
inputs: messages,
|
42 |
parameters: {
|
43 |
repetition_penalty: 1,
|
44 |
return_full_text: false
|
|
|
2 |
import { Configuration, OpenAIApi } from "openai-edge";
|
3 |
import { HfInference } from '@huggingface/inference';
|
4 |
import { HuggingFaceStream } from 'ai';
|
5 |
+
import { experimental_buildLlama2Prompt } from 'ai/prompts';
|
6 |
import { LLMStream } from "../../../utils/llm";
|
7 |
|
8 |
export const runtime = 'edge';
|
|
|
39 |
const response = Hf.textGenerationStream({
|
40 |
// @ts-ignore
|
41 |
model: aiProvider,
|
42 |
+
inputs: experimental_buildLlama2Prompt(messages),
|
43 |
parameters: {
|
44 |
repetition_penalty: 1,
|
45 |
return_full_text: false
|