Merge pull request #56 from chenyuan99/feature/add-deepseek-model-sel…

…ection
chenyuan99 · Feb 6, 2025 · 507510c · 507510c
2 parents 56db9d0 + 8e1f7ec
commit 507510c
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 7 deletions.
diff --git a/frontend/src/pages/JobGPT.tsx b/frontend/src/pages/JobGPT.tsx
@@ -1,16 +1,22 @@
 import React, { useState } from 'react';
 import { FileText, Upload, ThumbsUp } from 'lucide-react';
 import { jobgptService } from '../services/jobgptService';
-import { JobGPTMode, JobGPTState } from '../types/jobgpt';
+import { JobGPTMode, JobGPTState, DeepseekModel } from '../types/jobgpt';
 
 const initialState: JobGPTState = {
-  mode: 'why_company',  // Updated default mode
+  mode: 'why_company',
+  model: 'deepseek-coder-6.7b',
   isLoading: false,
   error: null,
   input: '',
   output: '',
 };
 
+const MODELS: DeepseekModel[] = [
+  'deepseek-coder-6.7b',
+  'deepseek-coder-33b'
+];
+
 export function JobGPT() {
   const [state, setState] = useState<JobGPTState>(initialState);
   const [selectedFile, setSelectedFile] = useState<File | null>(null);
@@ -19,6 +25,10 @@ export function JobGPT() {
     setState({ ...state, mode, input: '', output: '', error: null });
   };
 
+  const handleModelChange = (model: DeepseekModel) => {
+    setState({ ...state, model });
+  };
+
   const handleInputChange = (e: React.ChangeEvent<HTMLTextAreaElement>) => {
     setState({ ...state, input: e.target.value });
   };
@@ -32,8 +42,8 @@ export function JobGPT() {
     setState({ ...state, isLoading: true, error: null });
 
     try {
-      const response = await jobgptService.generatePrompt(state.input, state.mode);
-      setState({ ...state, isLoading: false, output: response.response });  // Updated to use response instead of prompt
+      const response = await jobgptService.generatePrompt(state.input, state.mode, state.model);
+      setState({ ...state, isLoading: false, output: response.response });  
     } catch (error: any) {
       setState({ ...state, isLoading: false, error: error.message });
     }
@@ -96,6 +106,21 @@ export function JobGPT() {
             General
           </button>
         </div>
+
+        <div className="mb-4">
+          <label className="block text-sm font-medium text-gray-700 mb-2">Model</label>
+          <select
+            className="w-full p-2 border rounded bg-white"
+            value={state.model}
+            onChange={(e) => handleModelChange(e.target.value as DeepseekModel)}
+          >
+            {MODELS.map(model => (
+              <option key={model} value={model}>
+                {model}
+              </option>
+            ))}
+          </select>
+        </div>
 
         <textarea
           className="w-full h-32 p-2 border rounded mb-4"

diff --git a/frontend/src/services/jobgptService.ts b/frontend/src/services/jobgptService.ts
@@ -1,14 +1,15 @@
 import axios from 'axios';
-import { JobGPTResponse, ResumeUploadResponse, JobGPTMode } from '../types/jobgpt';
+import { JobGPTResponse, ResumeUploadResponse, JobGPTMode, DeepseekModel } from '../types/jobgpt';
 
 const API_URL = import.meta.env.VITE_API_URL || 'http://localhost:8000';
 
 export const jobgptService = {
-  generatePrompt: async (prompt: string, mode: JobGPTMode): Promise<JobGPTResponse> => {
+  generatePrompt: async (prompt: string, mode: JobGPTMode, model: DeepseekModel): Promise<JobGPTResponse> => {
     try {
       const response = await axios.post(`${API_URL}/api/jobgpt/prompt`, {
         prompt,
-        mode
+        mode,
+        model
       });
       return response.data;
     } catch (error: any) {

diff --git a/frontend/src/types/jobgpt.ts b/frontend/src/types/jobgpt.ts
@@ -1,4 +1,5 @@
 export type JobGPTMode = 'why_company' | 'behavioral' | 'general';
+export type DeepseekModel = 'deepseek-coder-6.7b' | 'deepseek-coder-33b';
 
 export interface JobGPTResponse {
   response: string;
@@ -14,6 +15,7 @@ export interface ResumeUploadResponse {
 
 export interface JobGPTState {
   mode: JobGPTMode;
+  model: DeepseekModel;
   isLoading: boolean;
   error: string | null;
   input: string;