@@ -43,20 +43,26 @@ def _instrument(self, **kwargs):
4343 tracer_provider = kwargs .get ("tracer_provider" )
4444 tracer = get_tracer (__name__ , "" , tracer_provider )
4545 version = importlib .metadata .version ("weaviate-client" )
46-
47- for api_name , api_config in APIS .items ():
48- if api_config .get ("OPERATION" ) in ["query" , "generate" ]:
49- wrap_function_wrapper (
50- api_config ["MODULE" ],
51- api_config ["METHOD" ],
52- generic_query_patch (api_name , version , tracer ),
53- )
54- elif api_config .get ("OPERATION" ) == "create" :
55- wrap_function_wrapper (
56- api_config ["MODULE" ],
57- api_config ["METHOD" ],
58- generic_collection_patch (api_name , version , tracer ),
59- )
46+ try :
47+ for api_name , api_config in APIS .items ():
48+ if api_config .get ("OPERATION" ) in ["query" , "generate" ]:
49+ wrap_function_wrapper (
50+ api_config ["MODULE" ],
51+ api_config ["METHOD" ],
52+ generic_query_patch (api_name , version , tracer ),
53+ )
54+ elif api_config .get ("OPERATION" ) == "create" :
55+ print (
56+ api_config ["MODULE" ],
57+ api_config ["METHOD" ],
58+ )
59+ wrap_function_wrapper (
60+ api_config ["MODULE" ],
61+ api_config ["METHOD" ],
62+ generic_collection_patch (api_name , version , tracer ),
63+ )
64+ except ModuleNotFoundError as e :
65+ pass
6066
6167 def _instrument_module (self , module_name ):
6268 pass
0 commit comments