mistral

Author: arre-ankit

http-backend/package-lock.json

@@ -16,6 +16,7 @@
     "@google/generative-ai": "^0.21.0",
     "@langchain/community": "^0.3.31",
     "@langchain/core": "^0.3.40",
+    "@mistralai/mistralai": "^1.5.0",
     "@octokit/plugin-retry": "^7.1.4",
     "@octokit/plugin-throttling": "^9.4.0",
     "@prisma/client": "^6.4.0",

http-backend/package.json

@@ -67,7 +67,7 @@ router.post("/create", async (req,res): Promise<any> => {
         }
     })
 
-      // Enqueue the GitHub indexing job – this runs in the background.
+      //Enqueue the GitHub indexing job – this runs in the background.
       await qstashClient.publishJSON({
         url: `https://pullscan.onrender.com/v1/api/background/index-github`,
         body: {

http-backend/src/router/project.ts

@@ -1,7 +1,7 @@
 import {GoogleGenerativeAI} from "@google/generative-ai"
-import { Document } from "@langchain/core/documents"
 import { commit_main_summary_prompt, pr_summary_prompt } from "../prompt"
-
+import fs from 'fs'; // Add this import at the top
+import {summary_pr_mistral} from "./mistral"
 
 export interface CodeChange {
     filename: string;
@@ -27,67 +27,35 @@ export const aisummariseCommit = async (diff: string,commitMessage:string) => {
 }
 
 export const aisummarisePR = async (codes:CodeChange[],pullReqTitle: string, pullReqMessage:string) => {
-    const full_code:string = codes.map((code) => code.patch + code.filename).join('\n');
-
-    // const pr_prompt = pr_summary_prompt({
-    //         title:pullReqTitle, 
-    //         description:pullReqMessage,
-    //         file_diff:full_code
-    //    })
-
-    // const prompt: MessageParam[] = [{
-    //     role: 'user',
-    //     content: `${pr_prompt}`
-    // }]
-
-    // const main_summary = await generateCompletionClaude(prompt)
-
-    const main_summary = await  await model.generateContent([
-        pr_summary_prompt({
-                    title:pullReqTitle, 
-                    description:pullReqMessage,
-                    file_diff:full_code
-               })
-        ])
-
-
-    return main_summary.response.text()
-}
-
+    const full_code:string = codes.map((code) => code.patch + code.filename).join('');
 
+    const system_prompt =  pr_summary_prompt({
+        title:pullReqTitle, 
+        description:pullReqMessage,
+        file_diff:full_code
+   })
+   const prompt = 'Summerise this PR'
 
-// This summarize code generate summary for the RAG 
-export const summarizecode = async (doc:Document) => {
-    console.log("getting summary for",doc.metadata.source);
-    try{
-        const code = doc.pageContent.slice(0,10000);
-        const response = await model.generateContent([
-            `You are an inteligen software engineer who specialize in onboarding junior software engineers on the project
-            You are onboaring juniors software engineer and explaing to them the purpose of the ${doc.metadata.source} file
-
-            Here is the code:
-            --------
-            ${code}
-            --------
-            Give a summary no more than 100 words of the code above`
-
-        ])
-
-        return response.response.text()
-
-    }catch(error){
-        return ''
-    }
-
-}
+    const main = await summary_pr_mistral(system_prompt,prompt)
+    return main
+    
+    // fs.writeFileSync(`'full_code_${i}.txt'`, full_code, 'utf8'); // Add this line to save the code
+    // i++;
+    // console.log("Title:", pullReqTitle)
+    // fs.writeFileSync(`'title_${i}.txt'`, pullReqTitle, 'utf8'); // Add this line to save the code
+    
+    // console.log("Description:",pullReqMessage)
+    // fs.writeFileSync(`'description_${i}.txt'`, pullReqMessage, 'utf8'); // Add this line to save the code
+    // console.log('code',full_code)
 
-export async function aigenerateEmbedding(summary:string) {
-    const model = genAI.getGenerativeModel({
-        model: "text-embedding-004"
-    })
+    // const main_summary = await model.generateContent([
+    //     pr_summary_prompt({
+    //                 title:pullReqTitle, 
+    //                 description:pullReqMessage,
+    //                 file_diff:full_code
+    //            })
+    //     ])
 
-    const result = await model.embedContent(summary)
-    const embedding = result.embedding
-    return embedding.values
-}
 
+    //return main_summary.response.text()
+}

http-backend/src/utils/ai/gemni.ts

@@ -0,0 +1,19 @@
+import { Mistral } from '@mistralai/mistralai';
+
+const apiKey = process.env.MISTRAL_API_KEY;
+
+const client = new Mistral({apiKey: apiKey});
+
+export const  summary_pr_mistral = async function(system_prompt:string,prompt:string) {
+    const chatResponse:any = await client.chat.complete({
+        model: 'mistral-small-latest',
+        messages: [{role: 'user', content: `${prompt}`},{role:'system',content:`${system_prompt}`}],
+    });
+
+    //console.log('Chat:', chatResponse.choices[0].message.content);
+
+    return  chatResponse?.choices[0]?.message?.content
+}
+
+
+

http-backend/src/utils/ai/mistral.ts

@@ -121,4 +121,4 @@ async function summarizePR(githubUrl: string, pullReqHashes: string, pr_number:n
 
 // }
 
-// main()
+// main();

http-backend/src/utils/github/pull-req.ts

@@ -1 +1 @@
-{"root":["./src/index.ts","./src/db/index.ts","./src/db/seed.ts","./src/queues/githubqueue.ts","./src/router/background.ts","./src/router/project.ts","./src/utils/prompt.ts","./src/utils/ai/claude.ts","./src/utils/ai/gemni.ts","./src/utils/ai/openai.ts","./src/utils/github/commit.ts","./src/utils/github/github-loader.ts","./src/utils/github/octakit.ts","./src/utils/github/pull-req.ts","./src/utils/langbase/memory.ts","./src/utils/langbase/pipe.ts","./src/utils/langbase/prompt.ts"],"version":"5.7.3"}
+{"root":["./src/index.ts","./src/db/index.ts","./src/db/seed.ts","./src/queues/githubqueue.ts","./src/router/background.ts","./src/router/project.ts","./src/utils/prompt.ts","./src/utils/ai/claude.ts","./src/utils/ai/gemni.ts","./src/utils/ai/mistral.ts","./src/utils/ai/openai.ts","./src/utils/github/commit.ts","./src/utils/github/github-loader.ts","./src/utils/github/octakit.ts","./src/utils/github/pull-req.ts","./src/utils/langbase/memory.ts","./src/utils/langbase/pipe.ts","./src/utils/langbase/prompt.ts"],"version":"5.7.3"}