# HG changeset patch # User bgruening # Date 1724408472 0 # Node ID 329bdb05dd90543826f473a560022671803e7cef # Parent db316ead4bda575bea91094e5a65262e97ca92b4 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/chatgpt commit d2d08c3866c0f4a2f10372ae15c5dac5ea2d0bf0 diff -r db316ead4bda -r 329bdb05dd90 chatgpt.py --- a/chatgpt.py Wed Aug 14 11:33:36 2024 +0000 +++ b/chatgpt.py Fri Aug 23 10:21:12 2024 +0000 @@ -40,26 +40,24 @@ vision_sup_ext = ["jpg", "jpeg", "png", "webp", "gif"] file_search_file_streams = [] -image_urls = [] +image_files = [] for path in context_files: ext = path.split(".")[-1].lower() - if ext in vision_sup_ext and model in ["gpt-4o", "gpt-4o-mini", "gpt-4-turbo"]: + if ext in vision_sup_ext: if os.path.getsize(path) > 20 * 1024 * 1024: - raise Exception(f"File {path} exceeds the 20MB limit and will not be processed.") + print(f"File {path} exceeds the 20MB limit and will not be processed.") + sys.exit(1) file = client.files.create(file=open(path, "rb"), purpose="vision") promt = {"type": "image_file", "image_file": {"file_id": file.id}} - image_urls.append(promt) - + image_files.append(promt) elif ext in file_search_sup_ext: file_search_file_streams.append(open(path, "rb")) - else: - raise Exception("Not supported file!") assistant = client.beta.assistants.create( - instructions="You are going to get question about the file(s).", + instructions="You will receive questions about files from file searches and image files. For file search queries, identify and retrieve the relevant files based on the question. For image file queries, analyze the image content and provide relevant information or insights based on the image data.", model=model, - tools=[{"type": "file_search"}] if file_search_file_streams else None, + tools=[{"type": "file_search"}] if file_search_file_streams else [], ) if file_search_file_streams: vector_store = client.beta.vector_stores.create() @@ -67,7 +65,8 @@ vector_store_id=vector_store.id, files=file_search_file_streams ) assistant = client.beta.assistants.update( - assistant_id=assistant.id, tool_resources={"file_search": {"vector_store_ids": [vector_store.id]}} + assistant_id=assistant.id, + tool_resources={"file_search": {"vector_store_ids": [vector_store.id]}}, ) messages = [ @@ -78,20 +77,24 @@ "type": "text", "text": question, }, - *image_urls, + *image_files, ], } ] thread = client.beta.threads.create(messages=messages) -run = client.beta.threads.runs.create_and_poll(thread_id=thread.id, assistant_id=assistant.id) -messages = list(client.beta.threads.messages.list(thread_id=thread.id, run_id=run.id)) +run = client.beta.threads.runs.create_and_poll( + thread_id=thread.id, assistant_id=assistant.id +) +assistant_messages = list( + client.beta.threads.messages.list(thread_id=thread.id, run_id=run.id) +) -message_content = messages[0].content[0].text.value +message_content = assistant_messages[0].content[0].text.value print("Output has been saved!") with open("output.txt", "w") as f: f.write(message_content) -for image in image_urls: +for image in image_files: client.files.delete(image["image_file"]["file_id"]) if file_search_file_streams: client.beta.vector_stores.delete(vector_store.id) diff -r db316ead4bda -r 329bdb05dd90 chatgpt.xml --- a/chatgpt.xml Wed Aug 14 11:33:36 2024 +0000 +++ b/chatgpt.xml Fri Aug 23 10:21:12 2024 +0000 @@ -2,26 +2,28 @@ Integrating OpenAI's ChatGPT into Galaxy 2024 - 0 + 1 python openai - - + + + + + + + + + + + + + + + + + + + + + + + + + - - - - - - -