add default model for intro example

crewAIInc · Nov 20, 2024 · 8069997 · 8069997
1 parent f4c42f6
commit 8069997
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/nvidia_models/intro/.env.example b/nvidia_models/intro/.env.example
@@ -0,0 +1,2 @@
+NVIDIA_API_KEY=
+MODEL=meta/llama-3.1-8b-instruct
diff --git 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
@@ -2,7 +2,7 @@
 name = "marketing_posts"
 version = "0.1.0"
 description = "marketing-posts using crewAI"
-authors = ["Your Name <you@example.com>"]
+authors = ["raspawar raspawar@nvidia.com"]
 
 [tool.poetry.dependencies]
 python = ">=3.10,<=3.13"
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		NVIDIA_API_KEY=
		MODEL=meta/llama-3.1-8b-instruct