Rename files

lexiforest · Apr 30, 2024 · 636552b · 636552b
1 parent 5a17cf9
commit 636552b
Show file tree

Hide file tree

Showing 15 changed files with 29 additions and 7 deletions.
diff --git a/ragbear/database/__init__.py → docs/models.rst b/ragbear/database/__init__.py → docs/models.rst
diff --git a/ragbear/_entry.py b/ragbear/_entry.py
@@ -1,9 +1,9 @@
 from contextlib import suppress
 from typing import List, Literal, Optional, TypedDict, Union
-from .query.prompts import format_query
-from .model import get_openai_client
-from .database.remote.duckduckgo import get_duckduckgo_results
 
+from .index.remote.duckduckgo import get_duckduckgo_results
+from .model import get_openai_client
+from .synthesis.prompts import format_query
 
 with suppress(ImportError):
     pass
@@ -20,6 +20,29 @@ class SourceVendorType(TypedDict):
 RewriteType = Literal["query2doc"]
 
 
+class Bear:
+
+    def __init__(
+        self,
+        model: str = "llama2",
+        base_url: str = "http://localhost:11434/v1/",
+        api_key: Optional[str] = "ollama",
+        source: Optional[SourceType] = None,
+        rerank: Optional[RerankType] = None,
+        rewrite: Optional[RewriteType] = None,
+        template: Optional[str] = None,
+        embedding_model: Optional[str] = None,
+    ): ...
+
+    def answer(self): ...
+
+    def __enter__(self): ...
+
+    def __exit__(self): ...
+
+    def __del__(self): ...
+
+
 def query(
     prompt: str,
     *,
@@ -30,7 +53,7 @@ def query(
     rerank: Optional[RerankType] = None,
     rewrite: Optional[RewriteType] = None,
     template: Optional[str] = None,
-    embedding_model: Optional[str] = None
+    embedding_model: Optional[str] = None,
 ):
     """Query the LLM, and return the summarized result.
 
@@ -51,12 +74,11 @@ def query(
     chat_completion = client.chat.completions.create(
         messages=[
             {
-                'role': 'user',
-                'content': query,
+                "role": "user",
+                "content": query,
             }
         ],
         model=model,
     )
-    # print(chat_completion)
 
     return chat_completion.choices[0].message.content
diff --git a/ragbear/database/index/__init__.py → ragbear/index/__init__.py b/ragbear/database/index/__init__.py → ragbear/index/__init__.py
diff --git a/ragbear/database/remote/__init__.py → ragbear/index/engines/__init__.py b/ragbear/database/remote/__init__.py → ragbear/index/engines/__init__.py
diff --git a/ragbear/database/index/hnsw.py → ragbear/index/engines/hnsw.py b/ragbear/database/index/hnsw.py → ragbear/index/engines/hnsw.py
diff --git a/ragbear/database/find.py → ragbear/index/find.py b/ragbear/database/find.py → ragbear/index/find.py
diff --git a/ragbear/database/ingest.py → ragbear/index/ingest.py b/ragbear/database/ingest.py → ragbear/index/ingest.py
diff --git a/ragbear/database/ingest/extract.py → ragbear/index/ingest/extract.py b/ragbear/database/ingest/extract.py → ragbear/index/ingest/extract.py
diff --git a/ragbear/database/vendor/__init__.py → ragbear/index/remote/__init__.py b/ragbear/database/vendor/__init__.py → ragbear/index/remote/__init__.py
diff --git a/ragbear/database/remote/duckduckgo.py → ragbear/index/remote/duckduckgo.py b/ragbear/database/remote/duckduckgo.py → ragbear/index/remote/duckduckgo.py
diff --git a/ragbear/database/schema.py → ragbear/index/schema.py b/ragbear/database/schema.py → ragbear/index/schema.py
diff --git a/ragbear/index/vendor/__init__.py b/ragbear/index/vendor/__init__.py
diff --git a/ragbear/database/vendor/wiki_html.py → ragbear/index/vendor/wiki_html.py b/ragbear/database/vendor/wiki_html.py → ragbear/index/vendor/wiki_html.py
diff --git a/ragbear/postprocess/__init__.py b/ragbear/postprocess/__init__.py
diff --git a/ragbear/query/prompts.py → ragbear/synthesis/prompts.py b/ragbear/query/prompts.py → ragbear/synthesis/prompts.py