OneKE / examples /config /NewsExtraction.yaml
ShawnRu's picture
update
e6e7506
model:
category: DeepSeek # model category, chosen from ChatGPT, DeepSeek, LLaMA, Qwen, ChatGLM, MiniCPM, OneKE.
model_name_or_path: deepseek-chat # model name, chosen from the model list of the selected category.
api_key: your_api_key # your API key for the model with API service. No need for open-source models.
base_url: https://api.deepseek.com # base URL for the API service. No need for open-source models.
extraction:
task: Base # task type, chosen from Base, NER, RE, EE.
instruction: Extract key information from the given text. # description for the task. No need for NER, RE, EE task.
use_file: true # whether to use a file for the input text. Default set to false.
file_path: ./data/input_files/Tulsi_Gabbard_News.html # path to the input file. No need if use_file is set to false.
output_schema: NewsReport # output schema for the extraction task. Selected the from schema repository.
mode: customized # extraction mode, chosen from quick, detailed, customized. Default set to quick. See src/config.yaml for more details.
update_case: false # whether to update the case repository. Default set to false.
show_trajectory: false # whether to display the extracted intermediate steps