apunkt/PageIndex
1
0
Fork 0
mirror of https://github.com/VectifyAI/PageIndex.git synced 2026-04-24 23:56:21 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Disable agent tracing and auto-add litellm/ prefix for retrieve_model

Browse source 
* Disable agent tracing and auto-add litellm/ prefix for retrieve_model

* Preserve supported retrieve_model prefixes

* Remove temporary retrieve_model tests

* Limit tracing disablement to demo execution
This commit is contained in:
Ray 2026-03-29 00:55:57 +08:00 • committed by GitHub
parent d50c293309
commit a108c021ae
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
examples/agentic_vectorless_rag_demo.py
View file

@ -24,7 +24,7 @@ import requests
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from agents import Agent, Runner, function_tool
from agents import Agent, Runner, function_tool, set_tracing_disabled
from agents.model_settings import ModelSettings
from agents.stream_events import RawResponsesStreamEvent, RunItemStreamEvent
from openai.types.responses import ResponseTextDeltaEvent, ResponseReasoningSummaryTextDeltaEvent
@ -135,6 +135,8 @@ def query_agent(client: PageIndexClient, doc_id: str, prompt: str, verbose: bool
if __name__ == "__main__":
set_tracing_disabled(True)
# Download PDF if needed
if not os.path.exists(PDF_PATH):
print(f"Downloading {PDF_URL} ...")