Spaces:
Sleeping
Sleeping
File size: 2,721 Bytes
55d9644 24c4def 55d9644 24c4def 55d9644 24c4def 55d9644 24c4def 55d9644 24c4def 55d9644 24c4def 55d9644 24c4def 55d9644 24c4def 55d9644 24c4def |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
import time
from pipeline.claim_generate import *
from pipeline.query_generate import *
from pipeline.tool_execute import *
from pipeline.judge import *
from pipeline.openai_wrapper import *
class Pipeline:
def __init__(self, type, api_key, base_url=None):
# 全局只实例化一个对象 会不会干扰prompt的结果
self.syncchat = SyncChat(model="gpt-4-1106-preview", api_key=api_key, base_url=base_url)
self.asyncchat = AsyncChat(model="gpt-4-1106-preview", api_key=api_key, base_url=base_url)
self.visionchat = VisionChat(model="gpt-4-vision-preview", api_key=api_key, base_url=base_url)
self.claim_generator = ClaimGenerator(prompt_path="pipeline/prompts/claim_generate.yaml",chat=self.syncchat)
self.query_generator = QueryGenerator(prompt_path="pipeline/prompts/query_generate.yaml",chat=self.asyncchat)
self.tool = Tool()
self.judger = Judger(prompt_path="pipeline/prompts/verification.yaml", chat=self.visionchat)
def run(self, text, image_path, type, use_attribue=False):
time1 = time.time()
response, claim_list = self.claim_generator.get_response(text=text)
time2 = time.time()
if use_attribue:
objects, attribute_ques_list, scenetext_ques_list, fact_ques_list = self.query_generator.get_response(claim_list=claim_list)
else:
objects, scenetext_ques_list, fact_ques_list = self.query_generator.get_response(claim_list=claim_list,type=type)
attribute_ques_list = None
time3 = time.time()
print(objects)
print(attribute_ques_list)
print(scenetext_ques_list)
print(fact_ques_list)
"""
需要定时清除文件
"""
object_res, attribue_res, text_res, fact_res = self.tool.execute(image_path=image_path,
new_path="pipeline/cache_files/",
objects=objects,
attribute_list=attribute_ques_list,
scenetext_list=scenetext_ques_list,
fact_list=fact_ques_list)
time4 = time.time()
response = self.judger.get_response(type, object_res, attribue_res, text_res, fact_res, claim_list, image_path)
time5 = time.time()
print("claim generate time:" + str(time2-time1))
print("query generate time:" + str(time3-time2))
print("tool execute time:" + str(time4-time3))
print("judge time:" + str(time5-time4))
return response,claim_list
|