mirror of
https://github.com/x1xhlol/system-prompts-and-models-of-ai-tools.git
synced 2026-08-16 21:24:09 +00:00
22 lines
512 B
Python
22 lines
512 B
Python
import re
|
|
import os
|
|
from utils.prompts import DOCUMENT_CLASSIFY_PROMPT
|
|
from llm_bedrock import model
|
|
|
|
|
|
def document_classifier(record_details):
|
|
|
|
FINAL_DOCUMENT_CLASSIFY_PROMPT = DOCUMENT_CLASSIFY_PROMPT.format(document=record_details)
|
|
|
|
json_string = model.invoke(FINAL_DOCUMENT_CLASSIFY_PROMPT).content
|
|
|
|
print(json_string)
|
|
|
|
pattern = r'<output>(.*?)</output>'
|
|
match = re.search(pattern, json_string, re.DOTALL)
|
|
|
|
if match:
|
|
return match.group(1).strip()
|
|
else:
|
|
return ""
|