ob/sprint
1
0
Fork 0
mirror of https://github.com/hex248/sprint.git synced 2026-02-08 02:33:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

take model for /ai/chat

Browse Source
This commit is contained in:
Oliver Bryan
2026-01-31 14:16:29 +00:00
parent 95beddaa6c
commit f6d74927d6
3 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
packages/backend/src/routes/ai/chat.ts
View File

@@ -10,7 +10,7 @@ export default async function aiChat(req: AuthedRequest) {
const parsed = parseQueryParams(url, ChatRequestSchema); const parsed = parseQueryParams(url, ChatRequestSchema);
if ("error" in parsed) return parsed.error; if ("error" in parsed) return parsed.error;
const { orgId, projectId, message } = parsed.data; const { orgId, projectId, message, model } = parsed.data;
const user = await getUserById(req.userId); const user = await getUserById(req.userId);
if (!user) { if (!user) {
@@ -20,7 +20,7 @@ export default async function aiChat(req: AuthedRequest) {
const context = await buildContext(orgId, projectId, user); const context = await buildContext(orgId, projectId, user);
const fullPrompt = `${SYSTEM_PROMPT}\n\n${context}\n\n<user_query>${message}</user_query>`; const fullPrompt = `${SYSTEM_PROMPT}\n\n${context}\n\n<user_query>${message}</user_query>`;
const response = await callAI(fullPrompt); const response = await callAI(fullPrompt, model);
return Response.json(response); return Response.json(response);
} }

2
packages/frontend/src/lib/query/hooks/chat.ts
View File

@@ -2,7 +2,7 @@ import type { ChatRequest, ChatResponse, ModelsResponse } from "@sprint/shared";
import { useMutation } from "@tanstack/react-query"; import { useMutation } from "@tanstack/react-query";
import { apiClient } from "@/lib/server"; import { apiClient } from "@/lib/server";
export function useChatMutation() { export function useChat() {
return useMutation<ChatResponse, Error, ChatRequest>({ return useMutation<ChatResponse, Error, ChatRequest>({
mutationKey: ["ai", "chat"], mutationKey: ["ai", "chat"],
mutationFn: async (input) => { mutationFn: async (input) => {

1
packages/shared/src/api-schemas.ts
View File

@@ -671,6 +671,7 @@ export const ChatRequestSchema = z.object({
orgId: z.coerce.number().int().positive("orgId must be a positive integer"), orgId: z.coerce.number().int().positive("orgId must be a positive integer"),
projectId: z.coerce.number().int().positive("projectId must be a positive integer"), projectId: z.coerce.number().int().positive("projectId must be a positive integer"),
message: z.string().min(1, "Message is required"), message: z.string().min(1, "Message is required"),
model: z.string().min(1, "Model is required"),
}); });
export type ChatRequest = z.infer<typeof ChatRequestSchema>; export type ChatRequest = z.infer<typeof ChatRequestSchema>;