diff --git a/pyproject.toml b/pyproject.toml index ae3071d..2bd338f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -11,7 +11,7 @@ authors = ["Owen Colegrove "] license = "Apache-2.0" readme = "README.md" name = 'sciphi-synthesizer' -version = '1.0.2' +version = '1.0.3' packages = [ { include = "synthesizer" } ] diff --git a/synthesizer/interface/llm/sciphi_interface.py b/synthesizer/interface/llm/sciphi_interface.py index d529236..776e79f 100644 --- a/synthesizer/interface/llm/sciphi_interface.py +++ b/synthesizer/interface/llm/sciphi_interface.py @@ -13,7 +13,6 @@ class SciPhiLLMInterface(LLMInterface): """A class to interface with the SciPhi API.""" provider_name = LLMProviderName.SCIPHI - system_message = "You are a helpful assistant." def __init__( self, @@ -32,21 +31,9 @@ def get_completion( logger.debug( f"Getting completion from SciPhi API for model={generation_config.model_name}" ) - if "instruct" in generation_config.model_name: - return self.model.get_instruct_completion( - prompt, generation_config - ) - else: - return self._model.get_chat_completion( - [ - { - "role": "system", - "content": SciPhiLLMInterface.system_message, - }, - {"role": "user", "content": prompt}, - ], - generation_config, - ) + return self.model.get_instruct_completion( + prompt, generation_config + ) def get_chat_completion( self, conversation: list[dict], generation_config: GenerationConfig diff --git a/synthesizer/llm/config_manager.py b/synthesizer/llm/config_manager.py index 031acb0..cff0c00 100644 --- a/synthesizer/llm/config_manager.py +++ b/synthesizer/llm/config_manager.py @@ -15,10 +15,6 @@ def get_config_for_provider( llm_provider_name: LLMProviderName, ) -> Type[LLMConfig]: """Get the configuration class for a given model.""" - print( - "LLMConfigManager.config_registry = ", - LLMConfigManager.config_registry, - ) config_class = LLMConfigManager.config_registry.get(llm_provider_name) if not config_class: