dmitry.galkin/chatbot-template
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add reasoning model (#750)

Browse source 
Co-authored-by: Matt Apperson <me@mattapperson.com>
This commit is contained in:
Jeremy 2025-02-03 20:33:15 +05:30 • committed by GitHub
parent 76804269c4
commit c61d4f91d4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
19 changed files with 342 additions and 209 deletions
Download patch file Download diff file Expand all files Collapse all files

7
lib/ai/tools/request-suggestions.ts
View file

@ -1,20 +1,17 @@
import { z } from 'zod';
import { Model } from '../models';
import { Session } from 'next-auth';
import { DataStreamWriter, streamObject, tool } from 'ai';
import { getDocumentById, saveSuggestions } from '@/lib/db/queries';
import { Suggestion } from '@/lib/db/schema';
import { customModel } from '..';
import { generateUUID } from '@/lib/utils';
import { myProvider } from '../models';
interface RequestSuggestionsProps {
model: Model;
session: Session;
dataStream: DataStreamWriter;
}
export const requestSuggestions = ({
model,
session,
dataStream,
}: RequestSuggestionsProps) =>
@ -39,7 +36,7 @@ export const requestSuggestions = ({
> = [];
const { elementStream } = streamObject({
model: customModel(model.apiIdentifier),
model: myProvider.languageModel('block-model'),
system:
'You are a help writing assistant. Given a piece of writing, please offer suggestions to improve the piece of writing and describe the change. It is very important for the edits to contain full sentences instead of just words. Max 5 suggestions.',
prompt: document.content,