scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

fix(typing/openai.py): fix openai typing error (version-related)

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-04 22:02:43 -07:00
parent 2deac08ff1
commit 66129bc921
1 changed files with 50 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
litellm/types/llms/openai.py
View file

@ -14,7 +14,6 @@ from openai.types.beta.threads.message_content import MessageContent
from openai.types.beta.threads.message import Message as OpenAIMessage
from openai.types.beta.thread_create_params import (
Message as OpenAICreateThreadParamsMessage,
ToolResources as OpenAICreateThreadParamsToolResources,
)
from openai.types.beta.assistant_tool_param import AssistantToolParam
from openai.types.beta.threads.run import Run
@ -53,6 +52,56 @@ class NotGiven:
NOT_GIVEN = NotGiven()
class ToolResourcesCodeInterpreter(TypedDict, total=False):
file_ids: List[str]
"""
A list of [file](https://platform.openai.com/docs/api-reference/files) IDs made
available to the `code_interpreter` tool. There can be a maximum of 20 files
associated with the tool.
"""
class ToolResourcesFileSearchVectorStore(TypedDict, total=False):
file_ids: List[str]
"""
A list of [file](https://platform.openai.com/docs/api-reference/files) IDs to
add to the vector store. There can be a maximum of 10000 files in a vector
store.
"""
metadata: object
"""Set of 16 key-value pairs that can be attached to a vector store.
This can be useful for storing additional information about the vector store in
a structured format. Keys can be a maximum of 64 characters long and values can
be a maxium of 512 characters long.
"""
class ToolResourcesFileSearch(TypedDict, total=False):
vector_store_ids: List[str]
"""
The
[vector store](https://platform.openai.com/docs/api-reference/vector-stores/object)
attached to this thread. There can be a maximum of 1 vector store attached to
the thread.
"""
vector_stores: Iterable[ToolResourcesFileSearchVectorStore]
"""
A helper to create a
[vector store](https://platform.openai.com/docs/api-reference/vector-stores/object)
with file_ids and attach it to this thread. There can be a maximum of 1 vector
store attached to the thread.
"""
class OpenAICreateThreadParamsToolResources(TypedDict, total=False):
code_interpreter: ToolResourcesCodeInterpreter
file_search: ToolResourcesFileSearch
class FileSearchToolParam(TypedDict, total=False):
type: Required[Literal["file_search"]]
"""The type of tool being defined: `file_search`"""