Skip to content
Merged
Show file tree
Hide file tree
Changes from 16 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion src/raglite/_chatml_function_calling.py
Original file line number Diff line number Diff line change
Expand Up @@ -400,7 +400,9 @@ def chatml_function_calling_with_streaming(

# Case 2: Automatic or fixed tool choice
# Case 2 step 1: Determine whether to respond with a message or a tool call
assert (isinstance(tool_choice, str) and tool_choice == "auto") or isinstance(tool_choice, dict)
assert (isinstance(tool_choice, str) and tool_choice in ("auto", "required")) or isinstance(
tool_choice, dict
)
if isinstance(tool_choice, dict):
tools = [t for t in tools if t["function"]["name"] == tool_choice["function"]["name"]]
assert tools
Expand Down
1 change: 1 addition & 0 deletions src/raglite/_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,3 +103,4 @@ class RAGLiteConfig:
# list[Chunk], or list[ChunkSpan].
search_method: SearchMethod = field(default=_vector_search, compare=False)
self_query: bool = False
agentic_iterations: int = 3
1 change: 0 additions & 1 deletion src/raglite/_litellm.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,6 @@

# Reduce the logging level for LiteLLM, flashrank, and httpx.
litellm.suppress_debug_info = True
litellm.drop_params = True # Drop unsupported parameters for models like GPT-5
os.environ["LITELLM_LOG"] = "WARNING"
logging.getLogger("LiteLLM").setLevel(logging.WARNING)
logging.getLogger("flashrank").setLevel(logging.WARNING)
Expand Down
Loading