logs: suppress verbose warnings (#8003)
* suppress nvidia warnings about nim key * Wrap import libs with the warnings handling * Handle sigterm for docker exists * makefile revert * [autofix.ci] apply automated fixes * ruff * use sqlmodel paginate * Try to update polling timeout --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
parent
fabf8641d2
commit
fe9aa9ecb3
13 changed files with 72 additions and 18 deletions
|
|
@ -4,7 +4,6 @@ from unittest.mock import patch
|
|||
|
||||
import openai
|
||||
import pytest
|
||||
from langchain_nvidia_ai_endpoints import ChatNVIDIA
|
||||
from langchain_openai import ChatOpenAI
|
||||
from langflow.components.helpers.structured_output import StructuredOutputComponent
|
||||
from langflow.helpers.base_model import build_model_from_schema
|
||||
|
|
@ -415,6 +414,12 @@ class TestStructuredOutputComponent(ComponentTestBaseWithoutClient):
|
|||
)
|
||||
def test_with_real_nvidia_model_simple_schema(self):
|
||||
# Create a real NVIDIA model
|
||||
try:
|
||||
from langchain_nvidia_ai_endpoints import ChatNVIDIA
|
||||
except ImportError as e:
|
||||
msg = "Please install langchain-nvidia-ai-endpoints to use the NVIDIA model."
|
||||
raise ImportError(msg) from e
|
||||
|
||||
llm = ChatNVIDIA(model="meta/llama-3.2-3b-instruct", temperature=0, max_tokens=10)
|
||||
|
||||
# Create a component with a simple schema
|
||||
|
|
|
|||
|
|
@ -164,7 +164,7 @@ async def test_build_flow_polling(client, json_memory_chatbot_no_llm, logged_in_
|
|||
self.status_code = codes.OK
|
||||
self.max_total_events = 50 # Limit to prevent infinite loops
|
||||
self.max_empty_polls = 10 # Maximum number of empty polls before giving up
|
||||
self.poll_timeout = 1.0 # Timeout for each polling request
|
||||
self.poll_timeout = 3.0 # Timeout for each polling request
|
||||
|
||||
async def aiter_lines(self):
|
||||
try:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue