File tree Expand file tree Collapse file tree 4 files changed +5
-8
lines changed Expand file tree Collapse file tree 4 files changed +5
-8
lines changed Original file line number Diff line number Diff line change @@ -90,14 +90,13 @@ impl LlmGenerationClient for Client {
9090 let encoded_api_key = encode ( & self . api_key ) ;
9191
9292 let resp = retryable:: run (
93- || async {
93+ || {
9494 self . client
9595 . post ( url)
9696 . header ( "x-api-key" , encoded_api_key. as_ref ( ) )
9797 . header ( "anthropic-version" , "2023-06-01" )
9898 . json ( & payload)
9999 . send ( )
100- . await
101100 } ,
102101 & retryable:: HEAVY_LOADED_OPTIONS ,
103102 )
Original file line number Diff line number Diff line change @@ -114,7 +114,7 @@ impl LlmGenerationClient for Client {
114114
115115 let url = self . get_api_url ( request. model , "generateContent" ) ;
116116 let resp = retryable:: run (
117- || async { self . client . post ( & url) . json ( & payload) . send ( ) . await } ,
117+ || self . client . post ( & url) . json ( & payload) . send ( ) ,
118118 & retryable:: HEAVY_LOADED_OPTIONS ,
119119 )
120120 . await ?;
@@ -173,7 +173,7 @@ impl LlmEmbeddingClient for Client {
173173 payload[ "taskType" ] = serde_json:: Value :: String ( task_type. into ( ) ) ;
174174 }
175175 let resp = retryable:: run (
176- || async { self . client . post ( & url) . json ( & payload) . send ( ) . await } ,
176+ || self . client . post ( & url) . json ( & payload) . send ( ) ,
177177 & retryable:: HEAVY_LOADED_OPTIONS ,
178178 )
179179 . await ?;
Original file line number Diff line number Diff line change @@ -66,12 +66,11 @@ impl LlmGenerationClient for Client {
6666 stream : Some ( false ) ,
6767 } ;
6868 let res = retryable:: run (
69- || async {
69+ || {
7070 self . reqwest_client
7171 . post ( self . generate_url . as_str ( ) )
7272 . json ( & req)
7373 . send ( )
74- . await
7574 } ,
7675 & retryable:: HEAVY_LOADED_OPTIONS ,
7776 )
Original file line number Diff line number Diff line change @@ -76,13 +76,12 @@ impl LlmEmbeddingClient for Client {
7676 }
7777
7878 let resp = retryable:: run (
79- || async {
79+ || {
8080 self . client
8181 . post ( url)
8282 . header ( "Authorization" , format ! ( "Bearer {}" , self . api_key) )
8383 . json ( & payload)
8484 . send ( )
85- . await
8685 } ,
8786 & retryable:: HEAVY_LOADED_OPTIONS ,
8887 )
You can’t perform that action at this time.
0 commit comments