Update knowledgeassistant/constant/pipeline/__init__.py

knowledgeassistant/constant/pipeline/__init__.py

@@ -1,19 +1,21 @@
-PIPELINE_NAME: str = "KnowledgeAssistant"
-ARTIFACT_DIR: str = "Artifacts"
-
-
-INPUT_TEXT_DIR: str = "input_text"
-INPUT_TEXT_FILE_NAME: str = "input_text.txt"
-
-
-DATA_SUMMARIZATION_DIR_NAME: str = "data_summarization"
-SUMMARIZED_TEXT_DIR: str = "summarized_text"
-SUMMARIZED_TEXT_FILE_NAME: str = "summarized_text.txt"
-
-KEYWORD_EXTRACTION_DIR_NAME: str = "keyword_extraction"
-EXTRACTED_KEYWORDS_DIR: str = "keywords"
-EXTRACTED_KEYWORDS_FILE_NAME: str = "keywords.txt"
-
-RAG_DIR_NAME: str = "RAG"
-RAG_DIR: str = "generation"
+import os
+
+PIPELINE_NAME: str = "KnowledgeAssistant"
+ARTIFACT_DIR: str = os.path.join("/tmp", "Artifacts")
+
+
+INPUT_TEXT_DIR: str = "input_text"
+INPUT_TEXT_FILE_NAME: str = "input_text.txt"
+
+
+DATA_SUMMARIZATION_DIR_NAME: str = "data_summarization"
+SUMMARIZED_TEXT_DIR: str = "summarized_text"
+SUMMARIZED_TEXT_FILE_NAME: str = "summarized_text.txt"
+
+KEYWORD_EXTRACTION_DIR_NAME: str = "keyword_extraction"
+EXTRACTED_KEYWORDS_DIR: str = "keywords"
+EXTRACTED_KEYWORDS_FILE_NAME: str = "keywords.txt"
+
+RAG_DIR_NAME: str = "RAG"
+RAG_DIR: str = "generation"
 RAG_FILE_NAME: str = "generated_text.txt"