litellm with aws bedrock (#206)

gyliu513 · Sep 16, 2024 · 488bc88 · 488bc88
1 parent fe1cbfb
commit 488bc88
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/litellm/litellm-bedrock.py b/litellm/litellm-bedrock.py
@@ -0,0 +1,12 @@
+
+from dotenv import load_dotenv
+load_dotenv()
+
+from litellm import completion
+
+response = completion(
+  model="bedrock/amazon.titan-text-express-v1",
+  messages=[{ "content": "Tell me sth about IBM?","role": "user"}]
+)
+
+print(response)