Eng 544 ai xplain sdk update llm functional tests to cover all new llm models (#248)

* Added recent model to LLM test

* Added combined models to test

* Passing in instantiated model

* Passing in instantiated model

---------

Co-authored-by: xainaz <[email protected]>

Author: xainaz

tests/functional/model/run_model_test.py

@@ -1,20 +1,31 @@
 __author__ = "thiagocastroferreira"
 
-import pytest
 
 from aixplain.enums import Function
 from aixplain.factories import ModelFactory
 from aixplain.modules import LLM
+from datetime import datetime, timedelta, timezone
+
+
+def pytest_generate_tests(metafunc):
+    if "llm_model" in metafunc.fixturenames:
+        four_weeks_ago = datetime.now(timezone.utc) - timedelta(weeks=4)
+        models = ModelFactory.list(function=Function.TEXT_GENERATION)["results"]
+
+        predefined_models = ["Groq Llama 3 70B", "Chat GPT 3.5", "GPT-4o", "GPT 4 (32k)"]
+        recent_models = [model for model in models if model.created_at and model.created_at >= four_weeks_ago]
+        combined_models = recent_models + [
+            ModelFactory.list(query=model, function=Function.TEXT_GENERATION)["results"][0] for model in predefined_models
+        ]
+        metafunc.parametrize("llm_model", combined_models)
 
 
-@pytest.mark.parametrize("llm_model", ["Groq Llama 3 70B", "Chat GPT 3.5", "GPT-4o", "GPT 4 (32k)"])
 def test_llm_run(llm_model):
     """Testing LLMs with history context"""
-    model = ModelFactory.list(query=llm_model, function=Function.TEXT_GENERATION)["results"][0]
 
-    assert isinstance(model, LLM)
+    assert isinstance(llm_model, LLM)
 
-    response = model.run(
+    response = llm_model.run(
         data="What is my name?",
         history=[{"role": "user", "content": "Hello! My name is Thiago."}, {"role": "assistant", "content": "Hello!"}],
     )