From 8069997b73e7b6aa56e922eae5c79be0c0e8c7f4 Mon Sep 17 00:00:00 2001 From: raspawar Date: Wed, 20 Nov 2024 12:29:49 +0530 Subject: [PATCH] add default model for intro example --- nvidia_models/intro/.env.example | 2 ++ nvidia_models/intro/main.py | 2 +- nvidia_models/marketing_strategy/pyproject.toml | 2 +- 3 files changed, 4 insertions(+), 2 deletions(-) create mode 100644 nvidia_models/intro/.env.example diff --git a/nvidia_models/intro/.env.example b/nvidia_models/intro/.env.example new file mode 100644 index 00000000..f3af9034 --- /dev/null +++ b/nvidia_models/intro/.env.example @@ -0,0 +1,2 @@ +NVIDIA_API_KEY= +MODEL=meta/llama-3.1-8b-instruct \ No newline at end of file diff --git a/nvidia_models/intro/main.py b/nvidia_models/intro/main.py index 9c5b84d7..676d5ebb 100644 --- a/nvidia_models/intro/main.py +++ b/nvidia_models/intro/main.py @@ -115,7 +115,7 @@ def set_callbacks(self, callbacks: List[Any]): litellm.callbacks = callbacks -model = os.environ.get("MODEL") +model = os.environ.get("MODEL", "meta/llama-3.1-8b-instruct") llm = ChatNVIDIA(model=model) default_llm = nvllm(model_str="nvidia_nim/" + model, llm=llm) diff --git a/nvidia_models/marketing_strategy/pyproject.toml b/nvidia_models/marketing_strategy/pyproject.toml index 4bfeba90..bbbac835 100644 --- a/nvidia_models/marketing_strategy/pyproject.toml +++ b/nvidia_models/marketing_strategy/pyproject.toml @@ -2,7 +2,7 @@ name = "marketing_posts" version = "0.1.0" description = "marketing-posts using crewAI" -authors = ["Your Name "] +authors = ["raspawar raspawar@nvidia.com"] [tool.poetry.dependencies] python = ">=3.10,<=3.13"