Skip to content

Commit

Permalink
Update agent development tutorial to use Together AI and Gemma 2.9B I…
Browse files Browse the repository at this point in the history
…T model.

PiperOrigin-RevId: 672576402
Change-Id: Ibe02c534d5c882f155894677cc15cd14e00bdf1d
  • Loading branch information
vezhnick authored and copybara-github committed Sep 9, 2024
1 parent 7315c76 commit 1456c9f
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions examples/tutorials/agent_development.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -114,9 +114,14 @@
"\n",
"Links to where to generate API Key\n",
"\n",
"Together AI = https://api.together.xyz/settings/api-keys \n",
"\n",
"Mistral = https://console.mistral.ai/api-keys/\n",
"\n",
"OpenAI = https://platform.openai.com/settings/profile?tab=api-keys (Profile -\u003e API Keys) Note: GPT-4o models can be unwilling to participate in game structures resulting in worse performance. Using API_TYPE=mistral, MODEL_NAME=codestral-latest is advised.\n",
"OpenAI = https://platform.openai.com/settings/profile?tab=api-keys (Profile -\u003e API Keys) Note: GPT-4o models can be unwilling to participate in game structures resulting in worse performance. \n",
"\n",
"\n",
"Using API_TYPE='together_ai', MODEL_NAME='google/gemma-2-9b-it' is advised.\n",
"\n",
"\n",
"## Disable language model \n",
Expand All @@ -142,14 +147,14 @@
"# @title Parameters (edit this cell)\n",
"\n",
"# Pick API_TYPE from concordia/language_model/utils.py, e.g. mistral.\n",
"API_TYPE = 'mistral'\n",
"API_TYPE = 'together_ai'\n",
"# Add your API key here or alternatively, leave this as None to get it from an\n",
"# environment variable.\n",
"API_KEY = '' # @param {\"type\":\"string\"}\n",
"API_KEY = 'YOUR API KEY HERE' # @param {\"type\":\"string\"}\n",
"# Pick a specific model e.g. gpt-4o if API_TYPE is openai, codestral-latest if\n",
"# API_TYPE is mistral. See the corresponding wrapper in concordia/language_model\n",
"# for links to the websites where the model names are listed for each API_TYPE.\n",
"MODEL_NAME = 'codestral-latest'\n",
"MODEL_NAME = 'google/gemma-2-9b-it'\n",
"# Select an embedder by specifying one of the sentence transformer embedding\n",
"# models listed at https://huggingface.co/sentence-transformers.\n",
"EMBEDDER_NAME = 'all-mpnet-base-v2'\n",
Expand Down

0 comments on commit 1456c9f

Please sign in to comment.