diff --git a/libs/langchain/langchain/llms/fireworks.py b/libs/langchain/langchain/llms/fireworks.py index 867c916c1a586..676c8813b066a 100644 --- a/libs/langchain/langchain/llms/fireworks.py +++ b/libs/langchain/langchain/llms/fireworks.py @@ -31,9 +31,9 @@ class Fireworks(BaseLLM): model: str = "accounts/fireworks/models/llama-v2-7b-chat" model_kwargs: dict = Field( default_factory=lambda: { - "temperature": 0.1, + "temperature": 0.7, "max_tokens": 512, - "top_p": 0.9, + "top_p": 1, }.copy() ) fireworks_api_key: Optional[str] = None diff --git a/libs/langchain/tests/integration_tests/chat_models/test_fireworks.py b/libs/langchain/tests/integration_tests/chat_models/test_fireworks.py index 9553c9f35c2da..2bb4409ec4e05 100644 --- a/libs/langchain/tests/integration_tests/chat_models/test_fireworks.py +++ b/libs/langchain/tests/integration_tests/chat_models/test_fireworks.py @@ -68,16 +68,6 @@ def test_chat_fireworks_llm_output_contains_model_id() -> None: assert llm_result.llm_output["model"] == chat.model -def test_chat_fireworks_llm_output_stop_words() -> None: - """Test llm_output contains model_id.""" - chat = ChatFireworks() - message = HumanMessage(content="Hello") - llm_result = chat.generate([[message]], stop=[","]) - assert llm_result.llm_output is not None - assert llm_result.llm_output["model"] == chat.model - assert llm_result.generations[0][0].text[-1] == "," - - def test_fireworks_invoke() -> None: """Tests chat completion with invoke""" chat = ChatFireworks() diff --git a/libs/langchain/tests/integration_tests/llms/test_fireworks.py b/libs/langchain/tests/integration_tests/llms/test_fireworks.py index dfddb1224e83f..c1068631ce15a 100644 --- a/libs/langchain/tests/integration_tests/llms/test_fireworks.py +++ b/libs/langchain/tests/integration_tests/llms/test_fireworks.py @@ -104,15 +104,6 @@ def test_fireworks_multiple_prompts() -> None: assert len(output.generations) == 2 -def test_fireworks_stop_words() -> None: - """Test completion with stop words.""" - llm = Fireworks() - output = llm.generate(["How is the weather in New York today?"], stop=[","]) - assert isinstance(output, LLMResult) - assert isinstance(output.generations, list) - assert output.generations[0][0].text[-1] == "," - - def test_fireworks_streaming() -> None: """Test stream completion.""" llm = Fireworks()