도구(web_search·file_search)를 사용하는 AI 앱 만들기

pip install -r requirements.txt

# import namespaces
from openai import OpenAI
from azure.identity import DefaultAzureCredential, get_bearer_token_provider

# Initialize the OpenAI client
token_provider = get_bearer_token_provider(
    DefaultAzureCredential(), "https://ai.azure.com/.default"
)

openai_client = OpenAI(
    base_url=azure_openai_endpoint,
    api_key=token_provider
)

# Create vector store and upload files
print("Creating vector store and uploading files...")
vector_store = openai_client.vector_stores.create(
    name="travel-brochures"
)
file_streams = [open(f, "rb") for f in glob.glob("brochures/*.pdf")]
if not file_streams:
    print("No PDF files found in the brochures folder!")
    return
file_batch = openai_client.vector_stores.file_batches.upload_and_poll(
    vector_store_id=vector_store.id,
    files=file_streams
)
for f in file_streams:
    f.close()
print(f"Vector store created with {file_batch.file_counts.completed} files.")

# Get a response using tools
response = openai_client.responses.create(
    model=model_deployment,
    instructions="""
    당신은 Margie's Travel에서 제공하는 여행 서비스 정보를 안내하는 여행 어시스턴트입니다.
    Answer questions about services offered by Margie's Travel using the provided travel brochures.
    Search the web for general information about destinations or current travel advice.
    """,
    input=input_text,
    previous_response_id=last_response_id,
    tools=[
        {
            "type": "file_search",
            "vector_store_ids": [vector_store.id]
        },
        {
            "type": "web_search"
        }
    ]
)
print(response.output_text)
last_response_id = response.id

az login

python tools-app.py