� FIX: RAG系统导入和API集成修复 - Fixed RAG imports, updated API endpoints and frontend integration

backend/api/routes.py

@@ -217,8 +217,8 @@ async def generate_text(request: PromptRequest):
         enhanced_system_prompt = request.system_prompt
         
         # Check if RAG is available and should be used
-        use_rag = getattr(request, 'use_rag', False) if hasattr(request, 'use_rag') else False
-        retrieval_count = getattr(request, 'retrieval_count', 3) if hasattr(request, 'retrieval_count') else 3
+        use_rag = request.use_rag or False
+        retrieval_count = request.retrieval_count or 3
         
         if RAG_AVAILABLE and use_rag:
             rag_system = get_rag_system()

backend/models.py

@@ -16,6 +16,9 @@ class PromptRequest(BaseModel):
     model_name: Optional[str] = None
     temperature: Optional[float] = 0.7
     max_new_tokens: Optional[int] = 1024
+    # RAG configuration
+    use_rag: Optional[bool] = False
+    retrieval_count: Optional[int] = 3
 
 
 class PromptResponse(BaseModel):

frontend/src/hooks/useChat.ts

@@ -128,7 +128,7 @@ export function useChat(options: UseChatOptions = {}) {
   }, [currentSessionId])
 
   // Send message
-  const sendMessage = useCallback(async (assistantInfo?: AssistantInfo) => {
+  const sendMessage = useCallback(async (assistantInfo?: AssistantInfo, ragConfig?: { useRag: boolean, retrievalCount: number }) => {
     if (!input.trim() || status.isLoading) return
 
     let sessionId = currentSessionId
@@ -176,7 +176,9 @@ export function useChat(options: UseChatOptions = {}) {
           system_prompt: systemPrompt || null,
           model_name: selectedModel,
           temperature,
-          max_new_tokens: maxTokens
+          max_new_tokens: maxTokens,
+          use_rag: ragConfig?.useRag || false,
+          retrieval_count: ragConfig?.retrievalCount || 3
         }),
       })
 

frontend/src/pages/Playground.tsx

@@ -90,6 +90,10 @@ export function Playground() {
     setMaxTokens
   } = useChat()
 
+  // RAG configuration state
+  const [ragEnabled, setRagEnabled] = useState(false)
+  const [retrievalCount, setRetrievalCount] = useState(3)
+
   // UI state - sidebar collapse states
   const [sessionsCollapsed, setSessionsCollapsed] = useState(false)
   const [configCollapsed, setConfigCollapsed] = useState(false)
@@ -623,7 +627,8 @@ export function Playground() {
                 e.preventDefault()
                 if (!selectedModel || !models.find(m => m.model_name === selectedModel)) return
                 const assistantInfo = getCurrentAssistantInfo()
-                await sendMessage(assistantInfo)
+                const ragConfig = { useRag: ragEnabled, retrievalCount }
+                await sendMessage(assistantInfo, ragConfig)
               }}
               isGenerating={isLoading}
               stop={stopGeneration}
@@ -718,6 +723,10 @@ export function Playground() {
                 <TabsContent value="documents" className="p-6 space-y-6 m-0 h-full overflow-y-auto">
                   <DocumentsTab 
                     isLoading={isLoading}
+                    ragEnabled={ragEnabled}
+                    setRagEnabled={setRagEnabled}
+                    retrievalCount={retrievalCount}
+                    setRetrievalCount={setRetrievalCount}
                   />
                 </TabsContent>
               </div>
@@ -1376,14 +1385,20 @@ function SystemInstructionsTab({
 
 // Documents Tab Component for RAG functionality
 function DocumentsTab({
-  isLoading
+  isLoading,
+  ragEnabled,
+  setRagEnabled,
+  retrievalCount,
+  setRetrievalCount
 }: {
   isLoading: boolean
+  ragEnabled: boolean
+  setRagEnabled: (enabled: boolean) => void
+  retrievalCount: number
+  setRetrievalCount: (count: number) => void
 }) {
   const [uploadedFiles, setUploadedFiles] = useState<any[]>([])
   const [isUploading, setIsUploading] = useState(false)
-  const [ragEnabled, setRagEnabled] = useState(false)
-  const [retrievalCount, setRetrievalCount] = useState(3)
 
   // Load existing documents on component mount
   useEffect(() => {
@@ -1418,27 +1433,21 @@ function DocumentsTab({
     if (!files) return
 
     setIsUploading(true)
-    console.log('Starting file upload process...', files.length, 'files')
     
     try {
       const formData = new FormData()
       
       for (const file of Array.from(files)) {
-        console.log('Adding file to FormData:', file.name, 'type:', file.type)
         formData.append('files', file)
       }
 
-      console.log('Making API request to /rag/upload...')
       const response = await fetch('/rag/upload', {
         method: 'POST',
         body: formData,
       })
 
-      console.log('Upload response status:', response.status)
-      
       if (response.ok) {
         const result = await response.json()
-        console.log('Upload result:', result)
         
         // Add successfully processed files to the list
         const newFiles = result.results
@@ -1453,19 +1462,15 @@ function DocumentsTab({
             chunks: r.chunks
           }))
         
-        console.log('Successfully processed files:', newFiles)
         setUploadedFiles(prev => [...prev, ...newFiles])
         
         // Show errors for failed uploads
         const failedUploads = result.results.filter((r: any) => !r.success)
         if (failedUploads.length > 0) {
           console.error('Some files failed to upload:', failedUploads)
-        } else {
-          console.log('All files uploaded successfully!')
         }
       } else {
-        const errorText = await response.text()
-        console.error('Upload failed:', response.status, response.statusText, errorText)
+        console.error('Upload failed:', response.statusText)
       }
       
       // Reset input

rag_system.py

@@ -17,9 +17,8 @@ try:
     from langchain_huggingface import HuggingFaceEmbeddings
     from langchain_community.vectorstores import FAISS
     from langchain.schema import Document
-except ImportError as e:
-    print(f"LangChain not installed or import error: {e}")
-    print("Install with: pip install langchain langchain-community langchain-huggingface pypdf python-docx faiss-cpu sentence-transformers")
+except ImportError:
+    print("LangChain not installed. Install with: pip install langchain langchain-community langchain-huggingface pypdf python-docx faiss-cpu sentence-transformers")
     
 class SimpleRAGSystem:
     def __init__(self):

static/index.html

@@ -5,7 +5,7 @@
     <link rel="icon" type="image/svg+xml" href="/vite.svg" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>Edge LLM</title>
-    <script type="module" crossorigin src="/assets/index-ea1ae8fd.js"></script>
+    <script type="module" crossorigin src="/assets/index-cf882c3b.js"></script>
     <link rel="stylesheet" href="/assets/index-4cff54ff.css">
   </head>
   <body>