try few things
This commit is contained in:
parent
fa5028a072
commit
8a082ad384
2 changed files with 13 additions and 11 deletions
16
.github/workflows/test_common.yml
vendored
16
.github/workflows/test_common.yml
vendored
|
|
@ -81,13 +81,7 @@ jobs:
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
run: poetry run pytest tests/
|
run: poetry run pytest tests/
|
||||||
|
|
||||||
# - name: Build with Poetry
|
|
||||||
# run: poetry build
|
|
||||||
#
|
|
||||||
# - name: Install Package
|
|
||||||
# run: |
|
|
||||||
# cd dist
|
|
||||||
# pip install *.whl
|
|
||||||
|
|
||||||
# - name: Download NLTK Punkt Tokenizer Models
|
# - name: Download NLTK Punkt Tokenizer Models
|
||||||
# run: |
|
# run: |
|
||||||
|
|
@ -102,6 +96,14 @@ jobs:
|
||||||
ENV: 'dev'
|
ENV: 'dev'
|
||||||
run: poetry run python ./cognee/tests/test_library.py
|
run: poetry run python ./cognee/tests/test_library.py
|
||||||
|
|
||||||
|
- name: Build with Poetry
|
||||||
|
run: poetry build
|
||||||
|
|
||||||
|
- name: Install Package
|
||||||
|
run: |
|
||||||
|
cd dist
|
||||||
|
pip install *.whl
|
||||||
|
|
||||||
# - run: |
|
# - run: |
|
||||||
# poetry run
|
# poetry run
|
||||||
# if: runner.os != 'Windows'
|
# if: runner.os != 'Windows'
|
||||||
|
|
|
||||||
|
|
@ -21,15 +21,15 @@ def get_llm_client():
|
||||||
|
|
||||||
if provider == LLMProvider.OPENAI:
|
if provider == LLMProvider.OPENAI:
|
||||||
from .openai.adapter import OpenAIAdapter
|
from .openai.adapter import OpenAIAdapter
|
||||||
return OpenAIAdapter(llm_config.llm_api_key, llm_config.llm_model)
|
return OpenAIAdapter(config.llm_api_key, config.llm_model)
|
||||||
elif provider == LLMProvider.OLLAMA:
|
elif provider == LLMProvider.OLLAMA:
|
||||||
from .generic_llm_api.adapter import GenericAPIAdapter
|
from .generic_llm_api.adapter import GenericAPIAdapter
|
||||||
return GenericAPIAdapter(llm_config.llm_endpoint, llm_config.llm_api_key, llm_config.llm_model, "Ollama")
|
return GenericAPIAdapter(config.llm_endpoint, config.llm_api_key, config.llm_model, "Ollama")
|
||||||
elif provider == LLMProvider.ANTHROPIC:
|
elif provider == LLMProvider.ANTHROPIC:
|
||||||
from .anthropic.adapter import AnthropicAdapter
|
from .anthropic.adapter import AnthropicAdapter
|
||||||
return AnthropicAdapter(llm_config.llm_model)
|
return AnthropicAdapter(config.llm_model)
|
||||||
elif provider == LLMProvider.CUSTOM:
|
elif provider == LLMProvider.CUSTOM:
|
||||||
from .generic_llm_api.adapter import GenericAPIAdapter
|
from .generic_llm_api.adapter import GenericAPIAdapter
|
||||||
return GenericAPIAdapter(llm_config.llm_endpoint, llm_config.llm_api_key, llm_config.llm_model, "Custom")
|
return GenericAPIAdapter(config.llm_endpoint, config.llm_api_key, config.llm_model, "Custom")
|
||||||
else:
|
else:
|
||||||
raise ValueError(f"Unsupported LLM provider: {provider}")
|
raise ValueError(f"Unsupported LLM provider: {provider}")
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue