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
|