File tree Expand file tree Collapse file tree 1 file changed +1
-3
lines changed Expand file tree Collapse file tree 1 file changed +1
-3
lines changed Original file line number Diff line number Diff line change @@ -132,7 +132,6 @@ async def _mock_completions_response(request) -> AsyncIterable[str]:
132132            assert  payload ["stream_options" ] ==  {"include_usage" : True }
133133            assert  payload ["prompt" ] is  not None 
134134            assert  len (payload ["prompt" ]) >  0 
135-             assert  payload ["max_completion_tokens" ] >  0 
136135            assert  payload ["max_tokens" ] >  0 
137136
138137            return  httpx .Response (  # type: ignore 
@@ -141,7 +140,7 @@ async def _mock_completions_response(request) -> AsyncIterable[str]:
141140                    type_ = "text" ,
142141                    prompt = payload ["prompt" ],
143142                    output_token_count = (
144-                         payload ["max_completion_tokens " ]
143+                         payload ["max_tokens " ]
145144                        if  payload .get ("ignore_eos" , False )
146145                        else  None 
147146                    ),
@@ -162,7 +161,6 @@ async def _mock_chat_completions_response(request):
162161            assert  payload ["messages" ] is  not None 
163162            assert  len (payload ["messages" ]) >  0 
164163            assert  payload ["max_completion_tokens" ] >  0 
165-             assert  payload ["max_tokens" ] >  0 
166164
167165            return  httpx .Response (  # type: ignore 
168166                200 ,
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments