phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: support postgresql inference store (#2310)
Some checks failed
Integration Auth Tests / test-matrix (oauth2_token) (push) Failing after 3s
Details
Integration Tests / test-matrix (http, post_training) (push) Failing after 11s
Details
Integration Tests / test-matrix (library, inference) (push) Failing after 13s
Details
Integration Tests / test-matrix (http, providers) (push) Failing after 15s
Details
Integration Tests / test-matrix (http, tool_runtime) (push) Failing after 16s
Details
Integration Tests / test-matrix (http, datasets) (push) Failing after 18s
Details
Integration Tests / test-matrix (http, scoring) (push) Failing after 16s
Details
Integration Tests / test-matrix (http, agents) (push) Failing after 19s
Details
Integration Tests / test-matrix (library, datasets) (push) Failing after 16s
Details
Integration Tests / test-matrix (http, inspect) (push) Failing after 18s
Details
Integration Tests / test-matrix (library, agents) (push) Failing after 18s
Details
Integration Tests / test-matrix (http, inference) (push) Failing after 20s
Details
Integration Tests / test-matrix (library, inspect) (push) Failing after 9s
Details
Integration Tests / test-matrix (library, post_training) (push) Failing after 10s
Details
Integration Tests / test-matrix (library, tool_runtime) (push) Failing after 8s
Details
Test External Providers / test-external-providers (venv) (push) Failing after 8s
Details
Integration Tests / test-matrix (library, scoring) (push) Failing after 9s
Details
Integration Tests / test-matrix (library, providers) (push) Failing after 11s
Details
Unit Tests / unit-tests (3.11) (push) Failing after 8s
Details
Unit Tests / unit-tests (3.10) (push) Failing after 8s
Details
Unit Tests / unit-tests (3.12) (push) Failing after 8s
Details
Unit Tests / unit-tests (3.13) (push) Failing after 8s
Details
Pre-commit / pre-commit (push) Successful in 57s
Details

Browse source 
# What does this PR do?
* Added support postgresql inference store
* Added 'oracle' template that demos how to config postgresql stores
(except for telemetry, which is not supported currently)


## Test Plan

llama stack build --template oracle --image-type conda --run
LLAMA_STACK_CONFIG=http://localhost:8321 pytest -s -v tests/integration/
--text-model accounts/fireworks/models/llama-v3p3-70b-instruct -k
'inference_store'
This commit is contained in:
ehhuang 2025-05-29 14:33:09 -07:00 • committed by GitHub
parent 168c7113df
commit 2603f10f95
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
32 changed files with 516 additions and 53 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/unit/utils/test_sqlstore.py
View file

@ -9,7 +9,7 @@ from tempfile import TemporaryDirectory
import pytest
from llama_stack.providers.utils.sqlstore.api import ColumnType
from llama_stack.providers.utils.sqlstore.sqlite.sqlite import SqliteSqlStoreImpl
from llama_stack.providers.utils.sqlstore.sqlalchemy_sqlstore import SqlAlchemySqlStoreImpl
from llama_stack.providers.utils.sqlstore.sqlstore import SqliteSqlStoreConfig
@ -17,7 +17,7 @@ from llama_stack.providers.utils.sqlstore.sqlstore import SqliteSqlStoreConfig
async def test_sqlite_sqlstore():
with TemporaryDirectory() as tmp_dir:
db_name = "test.db"
sqlstore = SqliteSqlStoreImpl(
sqlstore = SqlAlchemySqlStoreImpl(
SqliteSqlStoreConfig(
db_path=tmp_dir + "/" + db_name,
)