jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-15 05:50:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add PDF support for openai_conversation.generate_content service (#141588)

Browse Source 
Add PDF support for openai_conversation.generate_content service
This commit is contained in:
Denis Shulyaka
2025-03-28 03:05:54 +03:00
committed by GitHub
parent 31479056ed
commit 195919b5fb
3 changed files with 45 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/components/openai_conversation/test_init.py
View File

@@ -262,6 +262,27 @@ async def test_init_error(
},
0,
),
(
{"prompt": "Picture of a dog", "filenames": ["/a/b/c.pdf"]},
{
"input": [
{
"content": [
{
"type": "input_text",
"text": "Picture of a dog",
},
{
"type": "input_file",
"file_data": "data:application/pdf;base64,BASE64IMAGE1",
"filename": "/a/b/c.pdf",
},
],
},
],
},
1,
),
(
{"prompt": "Picture of a dog", "filenames": ["/a/b/c.jpg"]},
{
@@ -415,8 +436,8 @@ async def test_generate_content_service(
[True, False],
),
(
{"prompt": "Not a picture of a dog", "filenames": ["/a/b/c.pdf"]},
"Only images are supported by the OpenAI API,`/a/b/c.pdf` is not an image file",
{"prompt": "Not a picture of a dog", "filenames": ["/a/b/c.mov"]},
"Only images and PDF are supported by the OpenAI API,`/a/b/c.mov` is not an image file or PDF",
1,
[True],
[True],