bedrock test

2026-07-18 18:18:28 +00:00 · 2023-09-14 13:51:09 -07:00
parent ba3a5cee2e
commit b68e28fd05
1 changed files with 17 additions and 0 deletions
@@ -633,6 +633,23 @@ def test_completion_sagemaker():
    except Exception as e:
        pytest.fail(f"Error occurred: {e}")

+
+# def test_completion_bedrock():
+#     try:
+#         response = completion(
+#             model="bedrock/amazon.titan-tg1-large", 
+#             messages=messages,
+#             temperature=0.2,
+#             max_tokens=80,
+#             logger_fn=logger_fn
+#         )
+#         # Add any assertions here to check the response
+#         print(response)
+#     except Exception as e:
+#         pytest.fail(f"Error occurred: {e}")
+# test_completion_bedrock()
+
+
 # test_completion_sagemaker()
 ######## Test VLLM ########
 # def test_completion_vllm():