das-cli 1.0.6__py3-none-any.whl → 1.2.5__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- das/ai/plugins/dasai.py +11 -2
- das/cli.py +1234 -1110
- das/common/api.py +20 -0
- das/common/config.py +6 -0
- das/managers/digital_objects_manager.py +84 -0
- das/managers/download_manager.py +43 -1
- das/managers/entries_manager.py +37 -1
- das/managers/search_manager.py +17 -2
- das/services/digital_objects.py +142 -0
- das/services/downloads.py +101 -84
- das/services/entries.py +29 -3
- das_cli-1.2.5.dist-info/METADATA +1076 -0
- das_cli-1.2.5.dist-info/RECORD +32 -0
- {das_cli-1.0.6.dist-info → das_cli-1.2.5.dist-info}/WHEEL +1 -1
- das_cli-1.0.6.dist-info/METADATA +0 -408
- das_cli-1.0.6.dist-info/RECORD +0 -30
- {das_cli-1.0.6.dist-info → das_cli-1.2.5.dist-info}/entry_points.txt +0 -0
- {das_cli-1.0.6.dist-info → das_cli-1.2.5.dist-info}/licenses/LICENSE +0 -0
- {das_cli-1.0.6.dist-info → das_cli-1.2.5.dist-info}/top_level.txt +0 -0
das/ai/plugins/dasai.py
CHANGED
|
@@ -3,6 +3,7 @@ import logging
|
|
|
3
3
|
from dotenv import load_dotenv
|
|
4
4
|
import asyncio
|
|
5
5
|
from semantic_kernel import Kernel
|
|
6
|
+
from openai import OpenAI
|
|
6
7
|
from semantic_kernel.connectors.ai.open_ai import (
|
|
7
8
|
OpenAIChatCompletion,
|
|
8
9
|
)
|
|
@@ -22,9 +23,17 @@ class DasAI:
|
|
|
22
23
|
api_key = os.getenv("OPENAI_API_KEY") or load_openai_api_key()
|
|
23
24
|
if not api_key:
|
|
24
25
|
raise ValueError("OpenAI API key is not configured.")
|
|
25
|
-
self.openai_chat_completion = OpenAIChatCompletion(ai_model_id="
|
|
26
|
+
self.openai_chat_completion = OpenAIChatCompletion(ai_model_id=os.getenv("OPEN_AI_MODEL_ID"), api_key=api_key)
|
|
26
27
|
self.kernel.add_service(self.openai_chat_completion)
|
|
27
|
-
self.kernel.add_plugin(GetEntryByCodePlugin(), plugin_name="get_entry_by_code")
|
|
28
|
+
self.kernel.add_plugin(GetEntryByCodePlugin(), plugin_name="get_entry_by_code")
|
|
29
|
+
# self.__load_available_models()
|
|
30
|
+
|
|
31
|
+
def __load_available_models(self):
|
|
32
|
+
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY") or load_openai_api_key())
|
|
33
|
+
models = client.models.list()
|
|
34
|
+
for model in models:
|
|
35
|
+
print(model.id)
|
|
36
|
+
|
|
28
37
|
|
|
29
38
|
async def main(self):
|
|
30
39
|
history = ChatHistory()
|