瀏覽代碼

处理引用错误

tenggangren 2 年之前
父節點
當前提交
bed657661e
共有 3 個文件被更改,包括 9 次插入0 次删除
  1. 3 0
      api/api.py
  2. 3 0
      chains/local_doc_qa.py
  3. 3 0
      models/chatglm_llm.py

+ 3 - 0
api/api.py

@@ -1,3 +1,6 @@
+import sys
+sys.path.append("..") # 将父目录放入系统路径中
+
 from fastapi import FastAPI, Request, UploadFile, File
 from fastapi import FastAPI, Request, UploadFile, File
 from fastapi.responses import StreamingResponse
 from fastapi.responses import StreamingResponse
 import uvicorn, json, datetime, time
 import uvicorn, json, datetime, time

+ 3 - 0
chains/local_doc_qa.py

@@ -1,3 +1,6 @@
+import sys
+sys.path.append("..") # 将父目录放入系统路径中
+
 from langchain.chains import RetrievalQA
 from langchain.chains import RetrievalQA
 from langchain.prompts import PromptTemplate
 from langchain.prompts import PromptTemplate
 from langchain.embeddings.huggingface import HuggingFaceEmbeddings
 from langchain.embeddings.huggingface import HuggingFaceEmbeddings

+ 3 - 0
models/chatglm_llm.py

@@ -1,3 +1,6 @@
+import sys
+sys.path.append("..") # 将父目录放入系统路径中
+
 from langchain.llms.base import LLM
 from langchain.llms.base import LLM
 from typing import Optional, List
 from typing import Optional, List
 from langchain.llms.utils import enforce_stop_tokens
 from langchain.llms.utils import enforce_stop_tokens