chore: have tests more robusts

Author: samuelint

Diff for: tests/test_functional/test_ainvoke.py

@@ -23,7 +23,7 @@ def llama(self, request) -> Llama:
 
     @pytest.fixture
     def instance(self, llama):
-        return LlamaChatModel(llama=llama)
+        return LlamaChatModel(llama=llama, temperature=0)
 
     @pytest.mark.asyncio
     async def test_ainvoke(self, instance: LlamaChatModel):
@@ -41,4 +41,4 @@ async def test_conversation_memory(self, instance: LlamaChatModel):
             ]
         )
 
-        assert "banana" in result.content
+        assert "banana" in result.content.lower()

Diff for: tests/test_functional/test_astream.py

@@ -16,7 +16,7 @@ def llama(self, request) -> Llama:
 
     @pytest.fixture
     def instance(self, llama):
-        return LlamaChatModel(llama=llama)
+        return LlamaChatModel(llama=llama, temperature=0)
 
     @pytest.mark.asyncio
     async def test_astream(self, instance: LlamaChatModel):
@@ -44,4 +44,4 @@ async def test_conversation_memory(self, instance: LlamaChatModel):
             final_content += token.content
 
         assert len(final_content) > 0
-        assert "banana" in final_content
+        assert "banana" in final_content.lower()

Diff for: tests/test_functional/test_invoke.py

@@ -24,7 +24,7 @@ def llama(self, request) -> Llama:
 
     @pytest.fixture
     def instance(self, llama):
-        return LlamaChatModel(llama=llama)
+        return LlamaChatModel(llama=llama, temperature=0)
 
     def test_invoke(self, instance: LlamaChatModel):
         result = instance.invoke("Say Hi!")
@@ -45,7 +45,7 @@ def test_conversation_memory(self, instance: LlamaChatModel):
             ]
         )
 
-        assert "banana" in result.content
+        assert "banana" in result.content.lower()
 
     def test_json_mode(self, instance: LlamaChatModel):
         structured_llm = instance.with_structured_output(Joke)

Diff for: tests/test_functional/test_llama_proxy.py

@@ -22,7 +22,9 @@ class TestLlamaProxyChat:
     )
     def instance(self, llama_proxy: LlamaProxy, request):
         return LlamaProxyChatModel(
-            llama_proxy=llama_proxy, model_name=request.param["alias"]
+            llama_proxy=llama_proxy,
+            model_name=request.param["alias"],
+            temperature=0,
         )
 
     def test_conversation_memory(self, instance: LlamaProxyChatModel):
@@ -34,4 +36,4 @@ def test_conversation_memory(self, instance: LlamaProxyChatModel):
             ]
         )
 
-        assert "banana" in result.content
+        assert "banana" in result.content.lower()