bump version to 0.1.14

.github/workflows/openai.yml

@@ -57,7 +57,7 @@ jobs:
         run: |
           pip install -e .[teachable]
       - name: Install packages for RetrieveChat with QDrant when needed
-        if: matrix.python-version == '3.9'
+        if: matrix.python-version == '3.11'
         run: |
           pip install qdrant_client[fastembed]
       - name: Coverage
@@ -80,6 +80,7 @@ jobs:
           OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
         run: |
           pip install nbconvert nbformat ipykernel
+          coverage run -a -m pytest test/agentchat/test_qdrant_retrievechat.py
           coverage run -a -m pytest test/test_with_openai.py
           coverage run -a -m pytest test/test_notebook.py
           coverage xml

autogen/retrieve_utils.py

@@ -1,9 +1,8 @@
-from typing import List, Union, Dict, Tuple, Callable
+from typing import List, Union, Callable
 import os
 import requests
 from urllib.parse import urlparse
 import glob
-import tiktoken
 import chromadb
 
 if chromadb.__version__ < "0.4.15":

autogen/version.py

@@ -1 +1 @@
-__version__ = "0.1.13"
+__version__ = "0.1.14"

test/oai/test_completion.py

@@ -231,7 +231,7 @@ def test_humaneval(num_samples=1):
         raise_on_ratelimit_or_timeout=False,
     )
     # assert response == -1
-    config_list = autogen.config_list_openai_aoai(KEY_LOC, exclude="aoai")
+    config_list = autogen.config_list_openai_aoai(KEY_LOC)
     # a minimal tuning example
     config, _ = autogen.Completion.tune(
         data=tune_data,
@@ -376,9 +376,9 @@ def test_math(num_samples=-1):
     ]
 
     autogen.Completion.set_cache(seed)
-    config_list = autogen.config_list_openai_aoai(KEY_LOC)[:2]
+    config_list = autogen.config_list_openai_aoai(KEY_LOC)
     vanilla_config = {
-        "model": "text-davinci-003",
+        "model": "text-ada-001",
         "temperature": 1,
         "max_tokens": 2048,
         "n": 1,
@@ -451,5 +451,5 @@ def my_average(results):
     # test_chatcompletion()
     # test_multi_model()
     # test_nocontext()
-    test_humaneval(1)
-    # test_math(1)
+    # test_humaneval(1)
+    test_math(1)

test/test_retrieve_utils.py

@@ -7,19 +7,15 @@
     extract_text_from_pdf,
     split_files_to_chunks,
     get_files_from_dir,
-    get_file_from_url,
     is_url,
     create_vector_db_from_dir,
     query_vector_db,
-    TEXT_FORMATS,
 )
 from autogen.token_count_utils import count_token
 
 import os
-import sys
 import pytest
 import chromadb
-import tiktoken
 
 
 test_dir = os.path.join(os.path.dirname(__file__), "test_files")
@@ -157,6 +153,7 @@ def custom_text_split_function(text):
             client=client,
             collection_name="mytestcollection",
             custom_text_split_function=custom_text_split_function,
+            get_or_create=True,
         )
         results = query_vector_db(["autogen"], client=client, collection_name="mytestcollection", n_results=1)
         assert (