lint all the things

nextcloud-releases · Jan 17, 2024 · ff3dfb4 · ff3dfb4
1 parent 273a644
commit ff3dfb4
Show file tree

Hide file tree

Showing 8 changed files with 19 additions and 33 deletions.
diff --git a/lib/chains.py b/lib/chains.py
@@ -1,5 +1,4 @@
-"""
-Registers all chains based on the models/ directory contents
+"""Registers all chains based on the models/ directory contents
 """
 
 import os
@@ -34,9 +33,9 @@ def generate_llm_generator(path):
 
 
 def generate_chains(model_name, model):
-    chains[model_name + ":summarize"] = lambda: SummarizeChain(llm=model())
-    chains[model_name + ":simplify"] = lambda: SimplifyChain(llm=model())
-    chains[model_name + ":formalize"] = lambda: FormalizeChain(llm=model())
+    chains[model_name + ":summary"] = lambda: SummarizeChain(llm=model())
+    #chains[model_name + ":simplify"] = lambda: SimplifyChain(llm=model())
+    #chains[model_name + ":formalize"] = lambda: FormalizeChain(llm=model())
     chains[model_name + ":headline"] = lambda: HeadlineChain(llm=model())
     chains[model_name + ":topics"] = lambda: TopicsChain(llm=model())
     chains[model_name + ":free_prompt"] = lambda: FreePromptChain(llm=model())

diff --git a/lib/formalize.py b/lib/formalize.py
@@ -1,5 +1,4 @@
-"""
-A langchain chain to formalize text
+"""A langchain chain to formalize text
 """
 
 from typing import Any, Optional
@@ -14,8 +13,7 @@
 
 
 class FormalizeChain(Chain):
-    """
-    A formalize chain
+    """A formalize chain
     """
 
     prompt: BasePromptTemplate = PromptTemplate(

diff --git a/lib/free_prompt.py b/lib/free_prompt.py
@@ -1,5 +1,4 @@
-"""
-A free rpompt chain
+"""A free rpompt chain
 """
 
 from typing import Any, Optional
@@ -12,8 +11,7 @@
 
 
 class FreePromptChain(Chain):
-    """
-    A free prompt chain
+    """A free prompt chain
     """
 
     llm: BaseLanguageModel

diff --git a/lib/headline.py b/lib/headline.py
@@ -1,5 +1,4 @@
-"""
-A chain to generate a headline for a text
+"""A chain to generate a headline for a text
 """
 
 from typing import Any, Optional
@@ -13,8 +12,7 @@
 
 
 class HeadlineChain(Chain):
-    """
-    A headline chain
+    """A headline chain
     """
 
     prompt: BasePromptTemplate = PromptTemplate(

diff --git a/lib/main.py b/lib/main.py
@@ -1,5 +1,4 @@
-"""
-Tha main module of the llm2 app
+"""Tha main module of the llm2 app
 """
 
 import queue
@@ -12,7 +11,7 @@
 from chains import chains
 from fastapi import Depends, FastAPI, responses
 from nc_py_api import AsyncNextcloudApp, NextcloudApp
-from nc_py_api.ex_app import LogLvl, anc_app
+from nc_py_api.ex_app import LogLvl, anc_app, run_app, set_handlers
 
 
 @asynccontextmanager
@@ -85,7 +84,7 @@ async def enabled_handler(enabled: bool, nc: AsyncNextcloudApp) -> str:
         for chain_name, _ in chains.items():
             (model, task) = chain_name.split(":", 2)
             await nc.providers.text_processing.register(
-                model, "Local Large language Model: " + model, "/chain/" + chain_name, task
+                "llm2:"+chain_name, "Local Large language Model: " + model, "/chain/" + chain_name, task
             )
     else:
         for chain_name, chain in chains.items():

diff --git a/lib/simplify.py b/lib/simplify.py
@@ -1,5 +1,4 @@
-"""
-A simplify chain
+"""A simplify chain
 """
 
 from typing import Any, Optional
@@ -14,8 +13,7 @@
 
 
 class SimplifyChain(Chain):
-    """
-    A summarization chain
+    """A summarization chain
     """
 
     prompt: BasePromptTemplate = PromptTemplate(

diff --git a/lib/summarize.py b/lib/summarize.py
@@ -1,5 +1,4 @@
-"""
-A recursive summarize chain
+"""A recursive summarize chain
 """
 
 from typing import Any, Optional
@@ -14,8 +13,7 @@
 
 
 class SummarizeChain(Chain):
-    """
-    A summarization chain
+    """A summarization chain
     """
 
     prompt: BasePromptTemplate = PromptTemplate(

diff --git a/lib/topics.py b/lib/topics.py
@@ -1,5 +1,4 @@
-"""
-A chain that extracts topcis from a text
+"""A chain that extracts topcis from a text
 """
 
 from typing import Any, Optional
@@ -13,8 +12,7 @@
 
 
 class TopicsChain(Chain):
-    """
-    A topics chain
+    """A topics chain
     """
 
     prompt: BasePromptTemplate = PromptTemplate(