minor hotfixes

threnjen · Dec 23, 2024 · 11daadc · 11daadc
1 parent d5c01ad
commit 11daadc
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 8 deletions.
diff --git a/Dockerfiles/Dockerfile.rag_description_generation b/Dockerfiles/Dockerfile.rag_description_generation
@@ -24,7 +24,7 @@ COPY modules/rag_description_generation modules/rag_description_generation
 COPY utils utils
 COPY modules/rag_description_generation/Pipfile* .
 COPY config* .
-COPY modules/rag_description_generation/nltk_data root/nltk_data
+COPY nltk_data root/nltk_data
 
 # Install dependencies with pipenv
 RUN pipenv sync

diff --git a/modules/rag_description_generation/rag_dynamodb.py b/modules/rag_description_generation/rag_dynamodb.py
@@ -17,8 +17,14 @@ def divide_and_process_generated_summary(self, game_id: str, summary: str) -> No
             .split("\n\n### Cons\n")[0]
             .replace("\n", "")
             .replace("-", "")
+            .strip()
+        )
+        cons = (
+            summary.split("\n\n### Cons\n")[-1]
+            .replace("\n", "")
+            .replace("-", "")
+            .strip()
         )
-        cons = summary.split("\n\n### Cons\n")[-1].replace("\n", "").replace("-", "")
 
         response = self.dynamodb_client.put_item(
             TableName="game_generated_descriptions",

diff --git a/modules/rag_description_generation/rag_weaviate.py b/modules/rag_description_generation/rag_weaviate.py
@@ -147,12 +147,12 @@ def create_weaviate_collection(self):
 
         self.weaviate_client.collections.create(
             name=self.collection_name,
-            # vectorizer_config=[
-            #     Configure.NamedVectors.text2vec_transformers(
-            #         name="title_vector",
-            #         source_properties=["title"],
-            #     )
-            # ],
+            vectorizer_config=[
+                Configure.NamedVectors.text2vec_transformers(
+                    name="title_vector",
+                    source_properties=["title"],
+                )
+            ],
             generative_config=wvc.config.Configure.Generative.openai(
                 model="gpt-4o-mini"
             ),