File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -85,7 +85,7 @@ def select_provider_type() -> Optional[ProviderType]:
8585
8686 # exact match only
8787 if result in provider_types :
88- return cast (ProviderType , result ) # type: ignore
88+ return cast (ProviderType , result )
8989
9090 # fuzzy match
9191 matches = [t for t in provider_types if result .lower () in t .lower ()]
Original file line number Diff line number Diff line change 2525try :
2626 import redis # type: ignore
2727except ImportError :
28- redis = None # type: ignore
28+ redis = None
2929
3030from nemoguardrails .rails .llm .config import EmbeddingsCacheConfig
3131
Original file line number Diff line number Diff line change @@ -46,7 +46,7 @@ class AzureEmbeddingModel(EmbeddingModel):
4646
4747 def __init__ (self , embedding_model : str ):
4848 try :
49- from openai import AzureOpenAI # type: ignore
49+ from openai import AzureOpenAI
5050 except ImportError :
5151 raise ImportError ("Could not import openai, please install it with `pip install openai`." )
5252 # Set Azure OpenAI API credentials
Original file line number Diff line number Diff line change @@ -42,7 +42,7 @@ class FastEmbedEmbeddingModel(EmbeddingModel):
4242 engine_name = "FastEmbed"
4343
4444 def __init__ (self , embedding_model : str , ** kwargs ):
45- from fastembed import TextEmbedding as Embedding # type: ignore
45+ from fastembed import TextEmbedding as Embedding
4646
4747 # Enabling a short form model name for all-MiniLM-L6-v2.
4848 if embedding_model == "all-MiniLM-L6-v2" :
Original file line number Diff line number Diff line change @@ -35,7 +35,7 @@ class NIMEmbeddingModel(EmbeddingModel):
3535
3636 def __init__ (self , embedding_model : str , ** kwargs ):
3737 try :
38- from langchain_nvidia_ai_endpoints import NVIDIAEmbeddings # type: ignore
38+ from langchain_nvidia_ai_endpoints import NVIDIAEmbeddings
3939
4040 self .model = embedding_model
4141 self .document_embedder = NVIDIAEmbeddings (model = embedding_model , ** kwargs )
Original file line number Diff line number Diff line change @@ -46,11 +46,11 @@ def __init__(
4646 ** kwargs ,
4747 ):
4848 try :
49- import openai # type: ignore
50- from openai import OpenAI # type: ignore
49+ import openai
50+ from openai import OpenAI
5151 except ImportError :
5252 raise ImportError ("Could not import openai, please install it with `pip install openai`." )
53- if openai .__version__ < "1.0.0" : # type: ignore
53+ if openai .__version__ < "1.0.0" :
5454 raise RuntimeError (
5555 "`openai<1.0.0` is no longer supported. Please upgrade using `pip install openai>=1.0.0`."
5656 )
Original file line number Diff line number Diff line change 6060if TYPE_CHECKING :
6161 from nemoguardrails .tracing import InteractionLog
6262try :
63- from opentelemetry import trace # type: ignore
64- from opentelemetry .trace import NoOpTracerProvider # type: ignore
63+ from opentelemetry import trace
64+ from opentelemetry .trace import NoOpTracerProvider
6565
6666except ImportError :
6767 raise ImportError (
@@ -100,7 +100,7 @@ def __init__(
100100 """
101101
102102 # validate that OpenTelemetry is properly configured
103- provider = trace .get_tracer_provider () # type: ignore
103+ provider = trace .get_tracer_provider ()
104104 if provider is None or isinstance (provider , NoOpTracerProvider ):
105105 warnings .warn (
106106 "No OpenTelemetry TracerProvider configured. Traces will not be exported. "
@@ -111,7 +111,7 @@ def __init__(
111111 stacklevel = 2 ,
112112 )
113113
114- self .tracer = trace .get_tracer ( # type: ignore
114+ self .tracer = trace .get_tracer (
115115 service_name ,
116116 instrumenting_library_version = version ("nemoguardrails" ),
117117 schema_url = "https://opentelemetry.io/schemas/1.26.0" ,
You can’t perform that action at this time.
0 commit comments