Skip to content

Commit ffaf21b

Browse files
committed
[VQA Extractor] Minor Fix
1 parent 8cb961c commit ffaf21b

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

docs/en/notes/guide/pipelines/PDFVQAExtractPipeline.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,8 @@ $env:DF_API_KEY = "sk-xxxxx"
5353
In the pipeline script, set your API endpoint:
5454
```python
5555
self.llm_serving = APILLMServing_request(
56-
api_url="https://api.openai.com/v1/chat/completions",
56+
api_url="https://generativelanguage.googleapis.com/v1beta/openai/chat/completions",
57+
key_name_of_api_key="DF_API_KEY",
5758
model_name="gemini-2.5-pro",
5859
max_workers=100,
5960
)
@@ -160,7 +161,7 @@ Example:
160161
```python
161162
from dataflow.serving import APILLMServing_request
162163
from dataflow.utils.storage import FileStorage
163-
from dataflow.operators.vqa import VQAExtractor
164+
from dataflow.operators.pdf2vqa import VQAExtractor
164165

165166
class VQA_extract_optimized_pipeline:
166167
def __init__(self):

docs/zh/notes/guide/pipelines/PDFVQAExtractPipeline.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,8 @@ $env:DF_API_KEY = "sk-xxxxx"
5353
在脚本中设置接口:
5454
```python
5555
self.llm_serving = APILLMServing_request(
56-
api_url="https://api.openai.com/v1/chat/completions",
56+
api_url="https://generativelanguage.googleapis.com/v1beta/openai/chat/completions",
57+
key_name_of_api_key="DF_API_KEY",
5758
model_name="gemini-2.5-pro",
5859
max_workers=100,
5960
)
@@ -159,7 +160,7 @@ self.storage = FileStorage(
159160
```python
160161
from dataflow.serving import APILLMServing_request
161162
from dataflow.utils.storage import FileStorage
162-
from dataflow.operators.vqa import VQAExtractor
163+
from dataflow.operators.pdf2vqa import VQAExtractor
163164

164165
class VQA_extract_optimized_pipeline:
165166
def __init__(self):

0 commit comments

Comments
 (0)