mixs_environments_extensions_to_envo_envirnmental_materials_by_claude…

….py works
microbiomedata · Mar 21, 2024 · 9be8f44 · 9be8f44
1 parent f216e54
commit 9be8f44
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 4 deletions.
diff --git a/nmdc_ontology/mixs_environments_extensions_to_envo_envirnmental_materials_by_claude.py b/nmdc_ontology/mixs_environments_extensions_to_envo_envirnmental_materials_by_claude.py
@@ -3,8 +3,14 @@
 
 from anthropic import Anthropic
 
-# Set up the Anthropic API client
+from dotenv import load_dotenv
+
+load_dotenv("../local/.env")
+
 api_key = os.environ["ANTHROPIC_API_KEY"]
+
+# Set up the Anthropic API client
+# api_key = os.environ["ANTHROPIC_API_KEY"]
 MODEL_NAME = "claude-3-opus-20240229"
 
 client = Anthropic(api_key=api_key)
@@ -35,9 +41,11 @@ def get_completion(client, prompt):
 completion = get_completion(client,
     f"""Here are the definitions of environments, according to MIxS: {mixs_environments} 
 and the definitions of environmental materials, according to EnvO: {envo_materials}.
-Generate an exhasutive YAML-formatted report of all environmental materials that could reasonalby be found in the Soil environment.
+Generate an exhaustive YAML-formatted report of all environmental materials 
+that could reasonably be found in the Soil environment.
 When associating an environmental material with an environment, 
-report both the environmental material id and the environmental material label for every associated environmental material.
+report both the environmental material id 
+and the environmental material label for every associated environmental material.
 """
 )
 

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -11,6 +11,7 @@ python = "^3.10"
 oaklib = "^0.5.25"
 pyarrow = "^15.0.0"
 anthropic = "^0.21.1"
+python-dotenv = "^1.0.1"
 
 [build-system]
 requires = ["poetry-core"]