Function streaming_query_loop

examples/gemini_example.py:78–108

Basic query loop for repeated questions

(rag: GraphRAG)

        raise
    
async def streaming_query_loop(rag: GraphRAG):
    """Basic query loop for repeated questions"""
    print("\nStreaming Query Interface (type 'exit' to quit)")
    print("="*50)
    
    while True:
        try:
            query = input("\nYou: ").strip()
            if query.lower() == 'exit':
                print("\nExiting chat...")
                break
                
            print("Assistant: ", end='', flush=True)
            
            try:
                # Higher token limits for Gemini context windows
                response = await rag.async_query(
                    query,
                    params=QueryParam(with_references=False, only_context=False, entities_max_tokens=250000, relations_max_tokens=200000, chunks_max_tokens=500000)
                )
                print(response.response)
            except Exception as e:
                import traceback
                traceback.print_exc()
                
        except KeyboardInterrupt:
            print("\nInterrupted by user")
            break
        except Exception as e:
            logger.exception("An error occurred:", exc_info=True)
            continue
        
### fast-graphrag example for Gemini
async def main():