Fix ai_complete test (#3612)

Author: sfc-gh-jdu

src/snowflake/snowpark/functions.py

@@ -12742,24 +12742,24 @@ def ai_complete(
         ... }
         >>> df = session.range(1).select(
         ...     ai_complete(
-        ...         model='mistral-large2',
+        ...         model='llama2-70b-chat',
         ...         prompt='Analyze the sentiment of this text: I love this product!',
         ...         response_format=response_schema
         ...     ).alias("structured_result")
         ... )
-        >>> result = df.collect()[0][0] # doctest: +SKIP
-        >>> 'sentiment' in result and 'confidence' in result # doctest: +SKIP
+        >>> result = df.collect()[0][0]
+        >>> 'sentiment' in result and 'confidence' in result
         True
 
         >>> # Using prompt object from prompt() function
         >>> df = session.range(1).select(
         ...     ai_complete(
-        ...         model='openai-gpt-4.1',
+        ...         model='claude-3-7-sonnet',
         ...         prompt=prompt("Extract the kitchen appliances identified in this image. Respond in JSON only with the identified appliances? {0}", to_file('@mystage/kitchen.png')),
         ...     )
         ... )
-        >>> result = df.collect()[0][0] # doctest: +SKIP
-        >>> "microwave" in result and "refrigerator" in result # doctest: +SKIP
+        >>> result = df.collect()[0][0]
+        >>> "microwave" in result and "refrigerator" in result
         True
     """
     sql_func_name = "ai_complete"