open-webui-automation/tools
2
0
Fork 0
Code Issues 9 Pull Requests Actions Packages Projects 1 Releases Wiki Activity

Update venice/chat.py

Browse Source
This commit is contained in:
xcaliber
2026-01-15 00:55:02 +00:00
parent 4ae3486b89
commit 0464abaa32
1 changed files with 14 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
venice/chat.py
View File

@@ -83,25 +83,15 @@ class Tools:
default="", description="Your Venice.ai API key (overrides admin default)" default="", description="Your Venice.ai API key (overrides admin default)"
) )
def __init__(self, valves: Optional[Dict] = None, __user__: dict = None, __request__: dict = None): def __init__(self):
"""Initialize with optional valve configuration from framework""" """Initialize with optional valve configuration from framework"""
# Handle valves configuration from framework # Handle valves configuration from framework
if valves:
self.valves = self.Valves(**valves)
else:
self.valves = self.Valves() self.valves = self.Valves()
# Enable tool usage visibility for debugging # Enable tool usage visibility for debugging
self.citation = True self.citation = True
# Handle user valves configuration # Handle user valves configuration
if __user__ and "valves" in __user__:
user_valves_dict = __user__.get("valves", {})
if isinstance(user_valves_dict, dict):
self.user_valves = self.UserValves(**user_valves_dict)
else:
self.user_valves = self.UserValves()
else:
self.user_valves = self.UserValves() self.user_valves = self.UserValves()
# Simple in-memory cache # Simple in-memory cache
@@ -137,10 +127,10 @@ class Tools:
return False return False
return (time.time() - self._cache_times[key]) < self.valves.MODEL_CACHE_TTL return (time.time() - self._cache_times[key]) < self.valves.MODEL_CACHE_TTL
def _format_error(self, e: Exception, context: str = "") -> str: def _format_error(self, e, context: str = "") -> str:
"""Format HTTP error with detailed context for LLM understanding.""" """Format HTTP error with detailed context for LLM understanding."""
try: try:
if hasattr(e, 'response') and e.response is not None: if hasattr(e, "response") and e.response is not None:
error_msg = e.response.text[:200] error_msg = e.response.text[:200]
try: try:
error_json = e.response.json() error_json = e.response.json()
@@ -180,7 +170,9 @@ class Tools:
pass pass
return {} return {}
async def _get_available_models(self, model_type: str = "text", __user__: dict = None) -> list[dict]: async def _get_available_models(
self, model_type: str = "text", __user__: dict = None
) -> list[dict]:
"""Fetch available models (cached).""" """Fetch available models (cached)."""
cache_key = f"models_{model_type}" cache_key = f"models_{model_type}"
if self._is_cache_valid(cache_key): if self._is_cache_valid(cache_key):
@@ -635,7 +627,11 @@ class Tools:
# Resolve and validate model (require reasoning capability) # Resolve and validate model (require reasoning capability)
resolved_model, error = await self._resolve_model( resolved_model, error = await self._resolve_model(
model, "text", require_reasoning=True, __model__=__model__, __user__=__user__ model,
"text",
require_reasoning=True,
__model__=__model__,
__user__=__user__,
) )
if error: if error:
return f"Error: {error}" return f"Error: {error}"