diff --git a/plugins/services/open_ai_client.py b/plugins/services/open_ai_client.py index fcb619d..6c789a7 100644 --- a/plugins/services/open_ai_client.py +++ b/plugins/services/open_ai_client.py @@ -6,7 +6,8 @@ load_dotenv() api_key = os.getenv("OPENAI_API_KEY") -client = OpenAI(api_key=api_key) +#client = OpenAI(api_key=api_key) +client = OpenAI(api_key=api_key, base_url="https://api.together.xyz/v1") def call_openai_api(model, prompt, llm_api_key=None): diff --git a/plugins/utils/auth.py b/plugins/utils/auth.py index 5336453..1335a9d 100644 --- a/plugins/utils/auth.py +++ b/plugins/utils/auth.py @@ -21,7 +21,8 @@ "embeddingModelAPIKey": os.getenv("EMBEDDING_MODEL_API_KEY"), "embeddingModel": os.getenv("EMBEDDING_MODEL"), "provider": "openai", - "model": "gpt-4o-mini", + # "model": "gpt-4o-mini", + "model": "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8", "apiKey": os.getenv("OPEN_AI_KEY"), "cohereKey": os.getenv("COHERE_KEY"), } diff --git a/requirements.txt b/requirements.txt index fe97f7f..0fab943 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,8 @@ alembic==1.13.2 annotated-types==0.7.0 anthropic==0.34.1 anyio==4.4.0 -apache-airflow==2.9.2 +apache-airflow[kubernetes,cncf.kubernetes]==2.9.2 +#apache-airflow==2.9.2 apache-airflow-providers-common-io==1.4.0 apache-airflow-providers-common-sql==1.17.1 apache-airflow-providers-fab==1.3.0 @@ -96,6 +97,7 @@ jsonpointer==3.0.0 jsonschema==4.23.0 jsonschema-specifications==2023.12.1 kiwisolver==1.4.7 +kubernetes langchain==0.2.15 langchain-community==0.2.6 langchain-core==0.2.36