TestFireworksStandard Class — langchain Architecture

Architecture documentation for the TestFireworksStandard class in test_standard.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  a301a024_f367_c036_79ae_4d954184da0e["TestFireworksStandard"]
  f4b40f56_1158_0e72_1cac_d0121e4eabd8["ChatModelUnitTests"]
  a301a024_f367_c036_79ae_4d954184da0e -->|extends| f4b40f56_1158_0e72_1cac_d0121e4eabd8
  f7c16a63_5839_40b5_57eb_c3ccfa5c4bba["test_standard.py"]
  a301a024_f367_c036_79ae_4d954184da0e -->|defined in| f7c16a63_5839_40b5_57eb_c3ccfa5c4bba
  7c9e3a5c_74f9_3e44_201c_e321235ce83e["chat_model_class()"]
  a301a024_f367_c036_79ae_4d954184da0e -->|method| 7c9e3a5c_74f9_3e44_201c_e321235ce83e
  7b077c02_122b_dbb0_d46f_3561b65a9476["chat_model_params()"]
  a301a024_f367_c036_79ae_4d954184da0e -->|method| 7b077c02_122b_dbb0_d46f_3561b65a9476
  5bbdb265_112f_6108_be1b_4c35a70b1b82["init_from_env_params()"]
  a301a024_f367_c036_79ae_4d954184da0e -->|method| 5bbdb265_112f_6108_be1b_4c35a70b1b82

Relationship Graph

Source Code

libs/partners/fireworks/tests/unit_tests/test_standard.py lines 11–37

class TestFireworksStandard(ChatModelUnitTests):
    @property
    def chat_model_class(self) -> type[BaseChatModel]:
        return ChatFireworks

    @property
    def chat_model_params(self) -> dict:
        return {
            "model": "accounts/fireworks/models/llama-v3p1-70b-instruct",
            "api_key": "test_api_key",
        }

    @property
    def init_from_env_params(self) -> tuple[dict, dict, dict]:
        return (
            {
                "FIREWORKS_API_KEY": "api_key",
                "FIREWORKS_API_BASE": "https://base.com",
            },
            {
                "model": "accounts/fireworks/models/llama-v3p1-70b-instruct",
            },
            {
                "fireworks_api_key": "api_key",
                "fireworks_api_base": "https://base.com",
            },
        )