更新代码

get_vector_db.py

@@ -7,7 +7,7 @@ COLLECTION_NAME = os.getenv('COLLECTION_NAME', 'siwei_ai')
 TEXT_EMBEDDING_MODEL = os.getenv('TEXT_EMBEDDING_MODEL', 'nomic-embed-text')
 
 def get_vector_db():
-    embedding = OllamaEmbeddings(model=TEXT_EMBEDDING_MODEL,show_progress=True)
+    embedding = OllamaEmbeddings(model=TEXT_EMBEDDING_MODEL,show_progress=True,num_gpu=0)
 
     db = Chroma(
         collection_name=COLLECTION_NAME,

query.py

@@ -21,7 +21,7 @@ def get_prompt():
     )
 
     template = """Answer the question in Chinese based ONLY on the following context:
-    {context}
+    {context} and output strictly in markdown format,Line breaks use <br>
     Question: {question}
     """
 

requirements1.txt

@@ -38,7 +38,7 @@ Flask==3.0.3
 flatbuffers==24.3.25
 fonttools==4.53.0
 frozenlist==1.4.1
-fsspec==2024.6.1
+fsspec==2024.2.0
 google-api-core==2.19.1
 google-auth==2.30.0
 google-cloud-vision==3.7.2