dinhquangson commited on
Commit
1cf8559
1 Parent(s): 46d4ef9

Update QueryMetadataExtractor.py

Browse files
Files changed (1) hide show
  1. QueryMetadataExtractor.py +19 -19
QueryMetadataExtractor.py CHANGED
@@ -11,28 +11,28 @@ class QueryMetadataExtractor:
11
 
12
  def __init__(self):
13
  prompt = """
14
- You are part of an information system that processes users queries.
15
- Given a user query you extract information from it that matches a given list of metadata fields.
16
- The information to be extracted from the query must match the semantics associated with the given metadata fields.
17
- The information that you extracted from the query will then be used as filters to narrow down the search space
18
- when querying an index.
19
- Just include the value of the extracted metadata without including the name of the metadata field.
20
- The extracted information in 'Extracted metadata' must be returned as a valid JSON structure.
21
  ###
22
- Example 1:
23
- Query: "What was the revenue of Nvidia in 2022?"
24
- Metadata fields: {"company", "year"}
25
- Extracted metadata fields: {"company": "nvidia", "year": 2022}
26
  ###
27
- Example 2:
28
- Query: "What were the most influential publications in 2023 regarding Alzheimer's disease?"
29
- Metadata fields: {"disease", "year"}
30
- Extracted metadata fields: {"disease": "Alzheimer", "year": 2023}
31
  ###
32
- Example 3:
33
- Query: "{{query}}"
34
- Metadata fields: "{{metadata_fields}}"
35
- Extracted metadata fields:
36
  """
37
  generator = OpenAIGenerator(
38
  api_key=Secret.from_env_var("OCTOAI_TOKEN"),
 
11
 
12
  def __init__(self):
13
  prompt = """
14
+ Bạn một phần của hệ thống thông tin xử lý các truy vấn của người dùng.
15
+ Với một truy vấn của người dùng, bạn trích xuất thông tin từ đó khớp với danh sách các trường siêu dữ liệu nhất định.
16
+ Thông tin được trích xuất từ ​​truy vấn phải khớp với ngữ nghĩa liên quan đến các trường siêu dữ liệu nhất định.
17
+ Thông tin bạn trích xuất từ ​​truy vấn sau đó sẽ được sử dụng làm bộ lọc để thu hẹp không gian tìm kiếm
18
+ khi truy vấn một chỉ mục.
19
+ Chỉ bao gồm giá trị của siêu dữ liệu được trích xuất không bao gồm tên của trường siêu dữ liệu.
20
+ Thông tin được trích xuất trong 'Siêu dữ liệu đã trích xuất' phải được trả về dưới dạng cấu trúc JSON hợp lệ.
21
  ###
22
+ dụ 1:
23
+ Truy vấn: "Doanh thu của Nvidia vào năm 2022 là bao nhiêu?"
24
+ Các trường siêu dữ liệu: {"công ty", "năm"}
25
+ Các trường siêu dữ liệu được trích xuất: {"company": "nvidia", "year": 2022}
26
  ###
27
+ dụ 2:
28
+ Truy vấn: "Ấn phẩm ảnh hưởng nhất vào năm 2023 liên quan đến bệnh Alzheimer là gì?"
29
+ Các trường siêu dữ liệu: {"bệnh", "năm"}
30
+ Các trường siêu dữ liệu được trích xuất: {"disease": "Alzheimer's", "year": 2023}
31
  ###
32
+ dụ 3:
33
+ Truy vấn: "{{truy vấn}}"
34
+ Trường siêu dữ liệu: "{{metadata_fields}}"
35
+ Các trường siêu dữ liệu được trích xuất:
36
  """
37
  generator = OpenAIGenerator(
38
  api_key=Secret.from_env_var("OCTOAI_TOKEN"),