feat:first commit

.gitignore

@@ -0,0 +1,164 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/latest/usage/project/#working-with-version-control
+.pdm.toml
+.pdm-python
+.pdm-build/
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+uv.lock
+.langgraph_api/

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 LangChain
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

Makefile

@@ -0,0 +1,67 @@
+.PHONY: all format lint test tests test_watch integration_tests docker_tests help extended_tests
+
+# Default target executed when no arguments are given to make.
+all: help
+
+# Define a variable for the test file path.
+TEST_FILE ?= tests/unit_tests/
+
+test:
+	python -m pytest $(TEST_FILE)
+
+integration_tests:
+	python -m pytest tests/integration_tests 
+
+test_watch:
+	python -m ptw --snapshot-update --now . -- -vv tests/unit_tests
+
+test_profile:
+	python -m pytest -vv tests/unit_tests/ --profile-svg
+
+extended_tests:
+	python -m pytest --only-extended $(TEST_FILE)
+
+
+######################
+# LINTING AND FORMATTING
+######################
+
+# Define a variable for Python and notebook files.
+PYTHON_FILES=src/
+MYPY_CACHE=.mypy_cache
+lint format: PYTHON_FILES=.
+lint_diff format_diff: PYTHON_FILES=$(shell git diff --name-only --diff-filter=d main | grep -E '\.py$$|\.ipynb$$')
+lint_package: PYTHON_FILES=src
+lint_tests: PYTHON_FILES=tests
+lint_tests: MYPY_CACHE=.mypy_cache_test
+
+lint lint_diff lint_package lint_tests:
+	python -m ruff check .
+	[ "$(PYTHON_FILES)" = "" ] || python -m ruff format $(PYTHON_FILES) --diff
+	[ "$(PYTHON_FILES)" = "" ] || python -m ruff check --select I $(PYTHON_FILES)
+	[ "$(PYTHON_FILES)" = "" ] || python -m mypy --strict $(PYTHON_FILES)
+	[ "$(PYTHON_FILES)" = "" ] || mkdir -p $(MYPY_CACHE) && python -m mypy --strict $(PYTHON_FILES) --cache-dir $(MYPY_CACHE)
+
+format format_diff:
+	ruff format $(PYTHON_FILES)
+	ruff check --select I --fix $(PYTHON_FILES)
+
+spell_check:
+	codespell --toml pyproject.toml
+
+spell_fix:
+	codespell --toml pyproject.toml -w
+
+######################
+# HELP
+######################
+
+help:
+	@echo '----'
+	@echo 'format                       - run code formatters'
+	@echo 'lint                         - run linters'
+	@echo 'test                         - run unit tests'
+	@echo 'tests                        - run unit tests'
+	@echo 'test TEST_FILE=<test_file>   - run all tests in file'
+	@echo 'test_watch                   - run unit tests in watch mode'
+

README.md

@@ -1,13 +1,80 @@
----
-title: ImobIA
-emoji: 👁
-colorFrom: gray
-colorTo: pink
-sdk: gradio
-sdk_version: 5.40.0
-app_file: app.py
-pinned: false
-short_description: primeira versão imobIA
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# New LangGraph Project
+
+[![CI](https://github.com/langchain-ai/new-langgraph-project/actions/workflows/unit-tests.yml/badge.svg)](https://github.com/langchain-ai/new-langgraph-project/actions/workflows/unit-tests.yml)
+[![Integration Tests](https://github.com/langchain-ai/new-langgraph-project/actions/workflows/integration-tests.yml/badge.svg)](https://github.com/langchain-ai/new-langgraph-project/actions/workflows/integration-tests.yml)
+
+This template demonstrates a simple application implemented using [LangGraph](https://github.com/langchain-ai/langgraph), designed for showing how to get started with [LangGraph Server](https://langchain-ai.github.io/langgraph/concepts/langgraph_server/#langgraph-server) and using [LangGraph Studio](https://langchain-ai.github.io/langgraph/concepts/langgraph_studio/), a visual debugging IDE.
+
+<div align="center">
+  <img src="./static/studio_ui.png" alt="Graph view in LangGraph studio UI" width="75%" />
+</div>
+
+The core logic defined in `src/agent/graph.py`, showcases an single-step application that responds with a fixed string and the configuration provided.
+
+You can extend this graph to orchestrate more complex agentic workflows that can be visualized and debugged in LangGraph Studio.
+
+## Getting Started
+
+<!--
+Setup instruction auto-generated by `langgraph template lock`. DO NOT EDIT MANUALLY.
+-->
+
+<!--
+End setup instructions
+-->
+
+1. Install dependencies, along with the [LangGraph CLI](https://langchain-ai.github.io/langgraph/concepts/langgraph_cli/), which will be used to run the server.
+
+```bash
+cd path/to/your/app
+pip install -e . "langgraph-cli[inmem]"
+```
+
+2. (Optional) Customize the code and project as needed. Create a `.env` file if you need to use secrets.
+
+```bash
+cp .env.example .env
+```
+
+If you want to enable LangSmith tracing, add your LangSmith API key to the `.env` file.
+
+```text
+# .env
+LANGSMITH_API_KEY=lsv2...
+```
+
+3. Start the LangGraph Server.
+
+```shell
+langgraph dev
+```
+
+For more information on getting started with LangGraph Server, [see here](https://langchain-ai.github.io/langgraph/tutorials/langgraph-platform/local-server/).
+
+## How to customize
+
+1. **Define configurable parameters**: Modify the `Configuration` class in the `graph.py` file to expose the arguments you want to configure. For example, in a chatbot application you may want to define a dynamic system prompt or LLM to use. For more information on configurations in LangGraph, [see here](https://langchain-ai.github.io/langgraph/concepts/low_level/?h=configuration#configuration).
+
+2. **Extend the graph**: The core logic of the application is defined in [graph.py](./src/agent/graph.py). You can modify this file to add new nodes, edges, or change the flow of information.
+
+## Development
+
+While iterating on your graph in LangGraph Studio, you can edit past state and rerun your app from previous states to debug specific nodes. Local changes will be automatically applied via hot reload.
+
+Follow-up requests extend the same thread. You can create an entirely new thread, clearing previous history, using the `+` button in the top right.
+
+For more advanced features and examples, refer to the [LangGraph documentation](https://langchain-ai.github.io/langgraph/). These resources can help you adapt this template for your specific use case and build more sophisticated conversational agents.
+
+LangGraph Studio also integrates with [LangSmith](https://smith.langchain.com/) for more in-depth tracing and collaboration with teammates, allowing you to analyze and optimize your chatbot's performance.
+
+<!--
+Configuration auto-generated by `langgraph template lock`. DO NOT EDIT MANUALLY.
+{
+  "config_schemas": {
+    "agent": {
+      "type": "object",
+      "properties": {}
+    }
+  }
+}
+-->

pyproject.toml

@@ -0,0 +1,65 @@
+[project]
+name = "agent"
+version = "0.0.1"
+description = "Starter template for making a new agent LangGraph."
+authors = [
+    { name = "William Fu-Hinthorn", email = "13333726+hinthornw@users.noreply.github.com" },
+]
+readme = "README.md"
+license = { text = "MIT" }
+requires-python = ">=3.9"
+dependencies = [
+    "langgraph>=0.2.6",
+    "python-dotenv>=1.0.1",
+]
+
+
+[project.optional-dependencies]
+dev = ["mypy>=1.11.1", "ruff>=0.6.1"]
+
+[build-system]
+requires = ["setuptools>=73.0.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools]
+packages = ["langgraph.templates.agent", "agent"]
+[tool.setuptools.package-dir]
+"langgraph.templates.agent" = "src/agent"
+"agent" = "src/agent"
+
+
+[tool.setuptools.package-data]
+"*" = ["py.typed"]
+
+[tool.ruff]
+lint.select = [
+    "E",    # pycodestyle
+    "F",    # pyflakes
+    "I",    # isort
+    "D",    # pydocstyle
+    "D401", # First line should be in imperative mood
+    "T201",
+    "UP",
+]
+lint.ignore = [
+    "UP006",
+    "UP007",
+    # We actually do want to import from typing_extensions
+    "UP035",
+    # Relax the convention by _not_ requiring documentation for every function parameter.
+    "D417",
+    "E501",
+]
+[tool.ruff.lint.per-file-ignores]
+"tests/*" = ["D", "UP"]
+[tool.ruff.lint.pydocstyle]
+convention = "google"
+
+[dependency-groups]
+dev = [
+    "anyio>=4.7.0",
+    "langgraph-cli[inmem]>=0.2.8",
+    "mypy>=1.13.0",
+    "pytest>=8.3.5",
+    "ruff>=0.8.2",
+]

src/agent/__init__.py

@@ -0,0 +1,8 @@
+"""New LangGraph Agent.
+
+This module defines a custom graph.
+"""
+
+from agent.graph import graph
+
+__all__ = ["graph"]

src/agent/agent_main.py

@@ -0,0 +1,81 @@
+from langchain.chat_models import ChatOpenAI
+from langchain.agents import AgentExecutor, tool
+from langchain.agents import initialize_agent
+from langchain.tools import Tool
+from langchain.chains import RetrievalQA
+from langchain.vectorstores import FAISS
+from langchain.embeddings import OpenAIEmbeddings
+from dotenv import load_dotenv
+import os
+import boto3
+
+
+load_dotenv()  # Carrega as variáveis do .env
+
+openai_api_key = os.getenv("OPENAI_API_KEY")
+aws_access_key_id = os.getenv("aws_access_key_id")
+aws_secret_access_key = os.getenv("aws_secret_access_key")
+
+
+def baixar_faiss_do_s3(bucket_name, local_dir, s3_prefix=""):
+    s3 = boto3.resource('s3',    
+                    aws_access_key_id = aws_access_key_id,
+                    aws_secret_access_key = aws_secret_access_key
+                    )
+    baldinho = s3.Bucket(bucket_name)
+    os.makedirs(local_dir, exist_ok=True)
+    for fname in ["index.faiss", "index.pkl"]:
+        s3_path = f"{s3_prefix}/{fname}" if s3_prefix else fname
+        local_path = os.path.join(local_dir, fname)
+        baldinho.download_file(s3_path, local_path)
+
+# Caminho temporário local
+VECTOR_DIR = "temp_faiss"
+
+# 🔁 Baixa do S3
+baixar_faiss_do_s3("imobia-app", VECTOR_DIR)
+
+# 🔗 Carrega o vetor FAISS a partir da pasta local
+embedding = OpenAIEmbeddings()
+vectorstore = FAISS.load_local(VECTOR_DIR, embedding,allow_dangerous_deserialization=True)
+
+# RAG tool
+@tool
+def consultar_empreendimentos(pergunta: str) -> str:
+    """Consulta a base de dados de empreendimentos imobiliários para responder perguntas sobre imóveis, preços, localização, metragem, diferenciais ou condições."""
+    chain = RetrievalQA.from_chain_type(
+        llm=ChatOpenAI(model="gpt-4o-mini", temperature=0),
+        retriever=vectorstore.as_retriever()
+    )
+    return chain.run(pergunta)
+
+tools = [consultar_empreendimentos]
+
+# LLM com suporte a function calling
+llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
+
+# Prompt do agente
+system_msg = """
+Você é um corretor de imóveis especializado em empreendimentos imobiliarios. Seja cordial e educado, sua função é 
+convencer o cliente a conhecer os empreendimentos e preparar a venda para o corretor.
+
+Sua função é ajudar o cliente com informações sobre localização, valores, metragem, diferenciais e condições de pagamento.
+
+Se a pergunta for sobre esses empreendimentos, use a ferramenta de consulta para responder com precisão.
+
+Nunca fale sobre assuntos fora do mercado imobiliário. Seja sempre educado, profissional e objetivo.
+
+No final de seu discurso sempre tente oferecer ajuda para entender melhor sobre os empreendimentos, codições de pagamento,
+seja útil e claro.
+"""
+
+agent = initialize_agent(
+    tools=tools,
+    llm=llm,
+    agent_type="openai-functions",
+    verbose=True,
+    agent_kwargs={"system_message": system_msg}
+)
+
+def responder_pergunta(pergunta):
+    return agent.run(pergunta)

src/agent/app.py

@@ -0,0 +1,16 @@
+import gradio as gr
+from agent_main import responder_pergunta
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🏡 imobIA")
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="Como posso te ajudar hoje?")
+    
+    def responder(chat_history, user_input):
+        resposta = responder_pergunta(user_input)
+        chat_history.append((user_input, resposta))
+        return chat_history, ""
+
+    msg.submit(responder, [chatbot, msg], [chatbot, msg])
+
+demo.launch()

tests/conftest.py

@@ -0,0 +1,6 @@
+import pytest
+
+
+@pytest.fixture(scope="session")
+def anyio_backend():
+    return "asyncio"

tests/integration_tests/__init__.py

@@ -0,0 +1 @@
+"""Define any integration tests you want in this directory."""

tests/integration_tests/test_graph.py

@@ -0,0 +1,12 @@
+import pytest
+
+from agent import graph
+
+pytestmark = pytest.mark.anyio
+
+
+@pytest.mark.langsmith
+async def test_agent_simple_passthrough() -> None:
+    inputs = {"changeme": "some_val"}
+    res = await graph.ainvoke(inputs)
+    assert res is not None

tests/unit_tests/__init__.py

@@ -0,0 +1 @@
+"""Define any unit tests you may want in this directory."""

tests/unit_tests/test_configuration.py

@@ -0,0 +1,9 @@
+from langgraph.pregel import Pregel
+
+from agent.graph import graph
+
+
+def test_placeholder() -> None:
+    # TODO: You can add actual unit tests
+    # for your graph and other logic here.
+    assert isinstance(graph, Pregel)