llms.mdx 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. ---
  2. title: 🤖 Large language models (LLMs)
  3. ---
  4. ## Overview
  5. Mem0 includes built-in support for various popular large language models. Memory can utilize the LLM provided by the user, ensuring efficient use for specific needs.
  6. <CardGroup cols={4}>
  7. <Card title="OpenAI" href="#openai"></Card>
  8. <Card title="Groq" href="#groq"></Card>
  9. </CardGroup>
  10. ## OpenAI
  11. To use OpenAI LLM models, you have to set the `OPENAI_API_KEY` environment variable. You can obtain the OpenAI API key from the [OpenAI Platform](https://platform.openai.com/account/api-keys).
  12. Once you have obtained the key, you can use it like this:
  13. ```python
  14. import os
  15. from mem0 import Memory
  16. os.environ['OPENAI_API_KEY'] = 'xxx'
  17. config = {
  18. "llm": {
  19. "provider": "openai",
  20. "config": {
  21. "model": "gpt-4o",
  22. "temperature": 0.2,
  23. "max_tokens": 1500,
  24. }
  25. }
  26. }
  27. m = Memory.from_config(config)
  28. m.add("Likes to play cricket on weekends", user_id="alice", metadata={"category": "hobbies"})
  29. ```
  30. ## Groq
  31. [Groq](https://groq.com/) is the creator of the world's first Language Processing Unit (LPU), providing exceptional speed performance for AI workloads running on their LPU Inference Engine.
  32. In order to use LLMs from Groq, go to their [platform](https://console.groq.com/keys) and get the API key. Set the API key as `GROQ_API_KEY` environment variable to use the model as given below in the example.
  33. ```python
  34. import os
  35. from mem0 import Memory
  36. os.environ['GROQ_API_KEY'] = 'xxx'
  37. config = {
  38. "llm": {
  39. "provider": "groq",
  40. "config": {
  41. "model": "mixtral-8x7b-32768",
  42. "temperature": 0.1,
  43. "max_tokens": 1000,
  44. }
  45. }
  46. }
  47. m = Memory.from_config(config)
  48. m.add("Likes to play cricket on weekends", user_id="alice", metadata={"category": "hobbies"})
  49. ```