client-openai-vector_stores-responses.py

Author: makelinux

llm/client-openai-vector_stores-responses.py

@@ -0,0 +1,23 @@
+#!/usr/bin/python3
+
+# ./llama-stack/docs/docs/getting_started/demo_script.py
+
+import io, requests
+from openai import OpenAI
+
+c = OpenAI(base_url="http://localhost:8321/v1/", api_key="none")
+
+vs = c.vector_stores.create()
+buf = io.BytesIO(str("abc is green").encode('utf-8'))
+
+fid = c.files.create(file=("aa.txt", buf), purpose="assistants").id
+c.vector_stores.files.create(vs.id, file_id=fid)
+
+resp = c.responses.create(
+    model="vertexai/google/gemini-2.5-flash",
+    input="What is the color of abc?",
+    tools=[{"type": "file_search", "vector_store_ids": [vs.id]}],
+#    include=["file_search_call.results"],
+)
+
+print(resp.output[-1].content[-1].text)