Updated README.md to clarify Pinecone account requirement, added newlines after command outputs, improved error handling for missing API keys, and updated version to 0.6.0: Co-authored-by: MindFlow <mf@mindflo.ai>

Author: Catherine

README.md

@@ -24,7 +24,7 @@ The [ChatGPT](https://openai.com/blog/chatgpt)-powered swiss army knife for the
 
 Pre-requisite: 
 - You'll need to create an [OpenAI](https://openai.com/blog/openai-api) account.
-- Also, create a [Pinecone](https://www.pinecone.io/start) account to use their vector database.
+- Also, create a [Pinecone](https://www.pinecone.io/start) account to use their vector database if you would like to use the `chat with documents` feature.
 
 1. Run `pip install mindflow`, or you can clone this repo and run `pip install -e path/to/mindflow`.
 2. Run `mf login`:

mindflow/__init__.py

@@ -1 +1 @@
-__version__ = "0.5.5"
+__version__ = "0.6.0"

mindflow/cli/commands/chat.py

@@ -41,6 +41,7 @@ async def stream_chat(settings: Settings, prompt: str):
                 click.echo(char_stream_chunk.value, nl=False)
             else:
                 click.echo(char_stream_chunk.value)
+        print("\n")
 
     async def stream_query(settings: Settings, file_paths: List[str], prompt: str):
         click.echo(colored("\nGPT:", attrs=["bold"]))
@@ -49,6 +50,7 @@ async def stream_query(settings: Settings, file_paths: List[str], prompt: str):
                 click.echo(char_stream_chunk.value, nl=False)
             else:
                 click.echo(char_stream_chunk.value)
+        print("\n")
 
     prompt, paths = parse_chat_prompt_and_paths_from_args(prompt_args)
     settings = Settings()
@@ -59,7 +61,7 @@ async def stream_query(settings: Settings, file_paths: List[str], prompt: str):
             )
         else:
             click.echo(
-                colored("Indexing paths... Note: this may take a while, if you want to skip this step, use the `--skip-index` flag. If you do so, you can pre-select specific paths to index with `mf index`.\n", "yellow")
+                colored("Indexing paths... Note: this may take a while, especially the first time. If it fails, or gets stuck, retry and it will pick up where you left off. If you want to skip this step, use the `--skip-index` flag. If you do so, you can pre-select specific paths to index with `mf index`.\n", "yellow")
             )
 
             asyncio.run(run_index(settings, paths))

mindflow/cli/commands/git/diff.py

@@ -39,5 +39,6 @@ async def stream_diff(settings: Settings, diff_output: str, detailed: bool):
                 click.echo(char_stream_chunk.value, nl=False)
             else:
                 click.echo(char_stream_chunk.value)
+        print("\n")
 
     asyncio.run(stream_diff(Settings(), diff_output, detailed=detailed))

mindflow/cli/commands/login.py

@@ -13,7 +13,7 @@ def login():
         ServiceConfigID.PINECONE.value,
     ]
     service_options = [
-        ServiceConfig.load(service_id, False) for service_id in service_ids
+        ServiceConfig.load(service_id) if ServiceConfig.load(service_id) is not None else ServiceConfig(service_id) for service_id in service_ids
     ]
     service_descriptions = ["OpenAI", "Pinecone: (Vector DB)"]
     service_config: ServiceConfig = select_option(

mindflow/core/types/model.py

@@ -1,4 +1,5 @@
 import json
+import sys
 import time
 import aiohttp
 import logging
@@ -189,8 +190,14 @@ def __init__(self, model_id: str):
                 }
             )
 
-        if service_config := ServiceConfig.load(f"{self.model.service}_config"):
-            self.api_key = service_config.api_key
+        service_config = ServiceConfig.load(f"{self.model.service}_config")
+        if service_config == None or not hasattr(service_config, "api_key"):
+            print(
+                f"No {self.model.service} API key found. Please set {self.model.service} API key with `mf login`."
+            )
+            sys.exit(1)
+        
+        self.api_key = service_config.api_key
 
         self.headers = {
             "Authorization": f"Bearer {self.api_key}",
@@ -399,7 +406,6 @@ async def call_api_stream(  # type: ignore
                                 await self.status_tracker.increment_requests_count_successful()
                                 await self.status_tracker.decrement_requests_count_in_progress()
                                 if not result["choices"][0]["delta"]:
-                                    print("No delta")
                                     continue
                                 yield Ok(
                                     result["choices"][0]["delta"]["content"]