gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

responses api (not streaming yet)

Browse source
This commit is contained in:
phact 2025-07-16 01:18:29 -04:00
parent 65b52ad1db
commit 4a2256a0f5
4 changed files with 20 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pyproject.toml
View file

@ -5,7 +5,7 @@ description = "Add your description here"
readme = "README.md" readme = "README.md"
requires-python = ">=3.13" requires-python = ">=3.13"
dependencies = [ dependencies = [
"agentd>=0.1.8", "agentd>=0.2.0.post2",
"aiofiles>=24.1.0", "aiofiles>=24.1.0",
"docling>=2.41.0", "docling>=2.41.0",
"opensearch-py[async]>=3.0.0", "opensearch-py[async]>=3.0.0",

22
src/agent.py
View file

@ -1,23 +1,17 @@
import asyncio
from agentd.patch import patch_openai_with_mcp
messages = [{"role": "system", "content": "You are a helpful assistant. use your tools to answer questions."}] messages = [{"role": "system", "content": "You are a helpful assistant. use your tools to answer questions."}]
# Async version for web server # Async version for web server
async def async_chat(async_client, prompt: str) -> str: async def async_chat(async_client, prompt: str, model: str = "gpt-4.1-mini", previous_response_id: str = None) -> str:
global messages global messages
messages += [{"role": "user", "content": prompt}] messages += [{"role": "user", "content": prompt}]
response = await async_client.chat.completions.create( response = await async_client.responses.create(
model="gpt-4.1-mini", model=model,
messages=messages, input=prompt,
mcp_strict=True previous_response_id=previous_response_id,
) )
response_text = response.choices[0].message.content response_id = response.id
response_text = response.output_text
print(f"user ==> {prompt}") print(f"user ==> {prompt}")
print(f"agent ==> {response_text}") print(f"agent ==> {response_text}")
return response_text return response_text
if __name__ == "__main__":
asyncio.run(async_chat("What pods are there?"))

10
src/app.py
View file

@ -12,6 +12,8 @@ import hashlib
import tempfile import tempfile
import asyncio import asyncio
import time import time
import json
import httpx
from starlette.applications import Starlette from starlette.applications import Starlette
from starlette.requests import Request from starlette.requests import Request
@ -107,7 +109,6 @@ async def init_index():
else: else:
print(f"Index '{INDEX_NAME}' already exists, skipping creation.") print(f"Index '{INDEX_NAME}' already exists, skipping creation.")
def extract_relevant(doc_dict: dict) -> dict: def extract_relevant(doc_dict: dict) -> dict:
""" """
Given the full export_to_dict() result: Given the full export_to_dict() result:
@ -310,11 +311,14 @@ app = Starlette(debug=True, routes=[
if __name__ == "__main__": if __name__ == "__main__":
import uvicorn import uvicorn
asyncio.run(init_index())
async def main():
await init_index()
asyncio.run(main())
uvicorn.run( uvicorn.run(
"app:app", "app:app",
host="0.0.0.0", host="0.0.0.0",
port=8000, port=8000,
reload=True, reload=True,
) )

8
uv.lock generated
View file

@ -9,7 +9,7 @@ resolution-markers = [
[[package]] [[package]]
name = "agentd" name = "agentd"
version = "0.1.8" version = "0.2.0.post2"
source = { registry = "https://pypi.org/simple" } source = { registry = "https://pypi.org/simple" }
dependencies = [ dependencies = [
{ name = "litellm" }, { name = "litellm" },
@ -18,9 +18,9 @@ dependencies = [
{ name = "openai-agents" }, { name = "openai-agents" },
{ name = "pyyaml" }, { name = "pyyaml" },
] ]
sdist = { url = "https://files.pythonhosted.org/packages/ea/e1/a3d0d2ddb6639be34d906f13a2edc91dbf174f8dcf97a68705f3a613ff8d/agentd-0.1.8.tar.gz", hash = "sha256:9278916d228d23d67283aed0e420d14f3b6862499df5cc5a8adb92ab3583ed17", size = 114252, upload-time = "2025-07-11T16:06:57.478Z" } sdist = { url = "https://files.pythonhosted.org/packages/91/fb/177ce5c7e8f8e8c4a4771b1da26e09e62780bf6f4042622654d05b101534/agentd-0.2.0.post2.tar.gz", hash = "sha256:b4cf8f5b727c1f0c0c9685762415e5affbc501758c0641eb9bd9c7d972c3ef30", size = 114513, upload-time = "2025-07-16T05:13:30.646Z" }
wheels = [ wheels = [
{ url = "https://files.pythonhosted.org/packages/d5/ca/9caa1253ab3ba151e725ea02aba334c29d659568b5341e5d886dbb394d85/agentd-0.1.8-py3-none-any.whl", hash = "sha256:15cc05ccbedfa9df8983a7a67c274c0c5a7ef029e55e6c0d7639106022c5cf06", size = 13472, upload-time = "2025-07-11T16:06:56.014Z" }, { url = "https://files.pythonhosted.org/packages/e4/5d/232b01286225d5e6eff6e4e741411950d60002eeeceb7b2ab1e84d96cc66/agentd-0.2.0.post2-py3-none-any.whl", hash = "sha256:000a058758843739061c93503db5977c9734f2e690545dd21923bf9ae8a8a161", size = 13266, upload-time = "2025-07-16T05:13:29.389Z" },
] ]
[[package]] [[package]]
@ -434,7 +434,7 @@ dependencies = [
[package.metadata] [package.metadata]
requires-dist = [ requires-dist = [
{ name = "agentd", specifier = ">=0.1.8" }, { name = "agentd", specifier = ">=0.2.0.post2" },
{ name = "aiofiles", specifier = ">=24.1.0" }, { name = "aiofiles", specifier = ">=24.1.0" },
{ name = "docling", specifier = ">=2.41.0" }, { name = "docling", specifier = ">=2.41.0" },
{ name = "opensearch-py", extras = ["async"], specifier = ">=3.0.0" }, { name = "opensearch-py", extras = ["async"], specifier = ">=3.0.0" },