Added github as dependency to allow using repo

zenml-io · Oct 31, 2024 · 64f3d8a · 64f3d8a
1 parent a7b30fc
commit 64f3d8a
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 1 deletion.
diff --git a/llm-complete-guide/configs/embeddings.yaml b/llm-complete-guide/configs/embeddings.yaml
@@ -26,6 +26,7 @@ settings:
       - rerankers[flashrank]
       - datasets
       - torch
+      - github
     environment:
       ZENML_PROJECT_SECRET_NAME: llm_complete
 

diff --git a/llm-complete-guide/configs/rag_eval.yaml b/llm-complete-guide/configs/rag_eval.yaml
@@ -12,6 +12,7 @@ settings:
       - numpy
       - psycopg2-binary
       - tiktoken
+      - github
 
 # configuration of the Model Control Plane
 model:

diff --git a/llm-complete-guide/configs/rag_gcp.yaml b/llm-complete-guide/configs/rag_gcp.yaml
@@ -12,6 +12,7 @@ settings:
       - tiktoken
       - ratelimit
       - rerankers
+      - github
     environment:
       ZENML_PROJECT_SECRET_NAME: llm_complete
       ZENML_ENABLE_RICH_TRACEBACK: FALSE

diff --git a/llm-complete-guide/configs/rag_local_dev.yaml b/llm-complete-guide/configs/rag_local_dev.yaml
@@ -14,6 +14,7 @@ settings:
       - tiktoken
       - ratelimit
       - rerankers
+      - github
     environment:
       ZENML_PROJECT_SECRET_NAME: llm_complete
       ZENML_ENABLE_RICH_TRACEBACK: FALSE

diff --git a/llm-complete-guide/configs/synthetic.yaml b/llm-complete-guide/configs/synthetic.yaml
@@ -24,6 +24,7 @@ settings:
       - datasets
       - torch
       - distilabel
+      - github
     environment:
       ZENML_PROJECT_SECRET_NAME: llm_complete
 

diff --git a/llm-complete-guide/run.py b/llm-complete-guide/run.py
@@ -208,7 +208,7 @@ def main(
         config_path = os.path.join(
             os.path.dirname(os.path.realpath(__file__)),
             "configs",
-            "rag_local_dev.yaml",
+            config,
         )
 
     if rag:
-Original file line number
+Diff line change
@@ Expand Up / @@ -12,6 +12,7 @@ settings: @@
           - numpy
           - psycopg2-binary
           - tiktoken
+          - github
     # configuration of the Model Control Plane
     model:
@@ Expand Down @@