Merge pull request #72 from leozqin/fix-summarization

Fix summarization

app/llm/ollama.py

@@ -29,6 +29,8 @@ def _make_chat_call(self, system: str, prompt: str):
         return chat["message"]["content"]
 
     def summarize(self, feed: Feed, entry: FeedEntry, mk: str):
-        system = self.system if self.system else self.system_prompt
+        system = self.system if self.system else self.summarization_system_prompt
 
-        return self._make_chat_call(system=system, prompt=self.get_prompt(mk))
+        return self._make_chat_call(
+            system=system, prompt=self.get_summarization_prompt(mk)
+        )

app/llm/openai.py

@@ -19,8 +19,8 @@ def summarize(self, feed: Feed, entry: FeedEntry, mk: str):
 
         completion = client.chat.completions.create(
             messages=[
-                {"role": "system", "content": self.system_prompt},
-                {"role": "user", "content": self.get_prompt(mk=mk)},
+                {"role": "system", "content": self.summarization_system_prompt},
+                {"role": "user", "content": self.get_summarization_prompt(mk=mk)},
             ],
             model=self.model,
             n=1,

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "precis"
-version = "0.3.1"
+version = "0.3.2"
 description = "A framework for automating your media diet"
 requires-python = ">=3.11"
 license = {file = "LICENSE"}