From c55035719903ff4f8b8d1f206d8a7e19338a5bf5 Mon Sep 17 00:00:00 2001
From: Kihaya Sugiura <kihayas@gmail.com>
Date: Sat, 20 Jan 2024 19:13:33 +0900
Subject: [PATCH] docs: Fix import

---
 docs/my-website/docs/providers/gemini.md | 4 ++--
 docs/my-website/docs/providers/palm.md   | 4 ++--
 docs/my-website/docs/providers/vertex.md | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/docs/my-website/docs/providers/gemini.md b/docs/my-website/docs/providers/gemini.md
index b50a850873..9d5eb298b8 100644
--- a/docs/my-website/docs/providers/gemini.md
+++ b/docs/my-website/docs/providers/gemini.md
@@ -6,7 +6,7 @@
 # Gemini-Pro
 ## Sample Usage
 ```python
-import litellm
+from litellm import completion
 import os
 
 os.environ['GEMINI_API_KEY'] = ""
@@ -24,7 +24,7 @@ LiteLLM Supports the following image types passed in `url`
 ## Sample Usage
 ```python
 import os
-import litellm 
+import litellm
 from dotenv import load_dotenv
 
 # Load the environment variables from .env file
diff --git a/docs/my-website/docs/providers/palm.md b/docs/my-website/docs/providers/palm.md
index 39fcc207c2..b41465b5fd 100644
--- a/docs/my-website/docs/providers/palm.md
+++ b/docs/my-website/docs/providers/palm.md
@@ -5,7 +5,7 @@
 
 ## Sample Usage
 ```python
-import litellm
+from litellm import completion
 import os
 
 os.environ['PALM_API_KEY'] = ""
@@ -17,7 +17,7 @@ response = completion(
 
 ## Sample Usage - Streaming
 ```python
-import litellm
+from litellm import completion
 import os
 
 os.environ['PALM_API_KEY'] = ""
diff --git a/docs/my-website/docs/providers/vertex.md b/docs/my-website/docs/providers/vertex.md
index f71aa0adab..3349faa529 100644
--- a/docs/my-website/docs/providers/vertex.md
+++ b/docs/my-website/docs/providers/vertex.md
@@ -17,7 +17,7 @@ import litellm
 litellm.vertex_project = "hardy-device-38811" # Your Project ID
 litellm.vertex_location = "us-central1"  # proj location
 
-response = completion(model="gemini-pro", messages=[{"role": "user", "content": "write code for saying hi from LiteLLM"}])
+response = litellm.completion(model="gemini-pro", messages=[{"role": "user", "content": "write code for saying hi from LiteLLM"}])
 ```
 
 ## Set Vertex Project & Vertex Location