请问各位前辈,我在调用chain.run后,余额不足,如何能捕获到该错误信息呢,代码如下,目前无法获取到错误信息
import langchain.llms.openai
from langchain import OpenAI, PromptTemplate, LLMChain
from langchain.text_splitter import CharacterTextSplitter
from langchain.chains.mapreduce import MapReduceChain
from langchain.prompts import PromptTemplate
from langchain.docstore.document import Document
from langchain.chains.summarize import load_summarize_chain
from langchain.document_loaders import UnstructuredFileLoader
from langchain.text_splitter import RecursiveCharacterTextSplitter
try:
text_splitter = CharacterTextSplitter()
loader = UnstructuredFileLoader("/public/quque/data/20230612143246198_summarize.txt")
# 将文本转成 Document 对象
document = loader.load()
# 初始化文本分割器
text_splitter = RecursiveCharacterTextSplitter(
chunk_size = 100,
chunk_overlap = 10,
)
#print(f'documents:{len(document)}')
# 切分文本
split_documents = text_splitter.split_documents(document)
#print(f'documents:{len(split_documents)}')
llm = OpenAI(openai_api_key="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",model_name="text-curie-001",max_tokens=1024,temperature=0.3)
print(llm)
prompt_template = """根据下面的内容做一下总结:
{text}
请使用中文回复:"""
PROMPT = PromptTemplate(template=prompt_template, input_variables=["text"])
# chain.run(docs)
chain = load_summarize_chain(llm, chain_type="refine", question_prompt=PROMPT, verbose=False)
#result = chain.run(split_documents[:len(split_documents)])
#print(result);
except Exception as e:
print("Error:", str(e))