Browse Source

处理引用错误

tenggangren 2 years ago
parent
commit
bed657661e
3 changed files with 9 additions and 0 deletions
  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.responses import StreamingResponse
 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.prompts import PromptTemplate
 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 typing import Optional, List
 from langchain.llms.utils import enforce_stop_tokens