phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-24 21:10:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add comprehensive unit tests for kvstore None handling fix

Browse source 
Signed-off-by: abhijeet-dhumal <abhijeetdhumal652@gmail.com>
This commit is contained in:
abhijeet-dhumal 2025-07-28 11:51:05 +05:30
parent ef3ed71875
commit 4e566276a5
No known key found for this signature in database
GPG key ID: 36718971C385EFFF
2 changed files with 223 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

9
llama_stack/providers/remote/vector_io/milvus/milvus.py
View file

@ -5,13 +5,20 @@
# the root directory of this source tree.
import asyncio
import json
import logging
import os
import re
from typing import Any
from numpy.typing import NDArray
from pymilvus import DataType, Function, FunctionType, MilvusClient
from pymilvus import DataType, MilvusClient
# Function and FunctionType are not available in all pymilvus versions
try:
from pymilvus import Function, FunctionType
except ImportError:
Function = None
FunctionType = None
from llama_stack.apis.files.files import Files
from llama_stack.apis.inference import Inference, InterleavedContent